Start testing lesana derivatives

1 files changed, 40 insertions, 0 deletions

diff --git a/tests/test_derivatives.py b/tests/test_derivatives.py
new file mode 100644
index 0000000..f79123c
--- /dev/null
+++ b/tests/test_derivatives.py
@@ -0,0 +1,40 @@
+import shutil
+import tempfile
+import unittest
+
+import lesana
+from lesana import types
+
+
+class DerivedType(types.LesanaString):
+    """
+    A custom type
+    """
+    name = 'derived'
+
+
+class Derivative(lesana.Collection):
+    """
+    A class serived from lesana.Collection
+    """
+
+
+class testDerivatives(unittest.TestCase):
+    def setUp(self):
+        self.tmpdir = tempfile.mkdtemp()
+        shutil.copytree(
+            'tests/data/derivative',
+            self.tmpdir,
+            dirs_exist_ok=True
+        )
+        self.collection = Derivative(self.tmpdir)
+
+    def tearDown(self):
+        shutil.rmtree(self.tmpdir)
+
+    def test_load_subclasses(self):
+        self.assertIsInstance(self.collection.fields['unknown'], DerivedType)
+
+
+if __name__ == '__main__':
+    unittest.main()