1 files changed, 30 insertions, 0 deletions
diff --git a/modelcif/test.py b/modelcif/test.py
new file mode 100644
index 0000000..843231c
--- /dev/null
+++ b/modelcif/test.py
@@ -0,0 +1,30 @@
+import modelcif
+import modelcif.dumper
+import modelcif.reader
+import os
+import unittest
+
+
+class Tests(unittest.TestCase):
+    def test_basic(self):
+        """Basic install test"""
+        system = modelcif.System(title='test system')
+
+        entity_a = modelcif.Entity('AAA', description='Subunit A')
+        entity_b = modelcif.Entity('AAAAAA', description='Subunit B')
+        system.entities.extend((entity_a, entity_b))
+
+        # Test output in mmCIF format
+        with open('output.cif', 'w') as fh:
+            modelcif.dumper.write(fh, [system])
+
+        # Make sure we can read back the file
+        with open('output.cif') as fh:
+            sys2, = modelcif.reader.read(fh)
+        self.assertEqual(sys2.title, 'test system')
+
+        os.unlink('output.cif')
+
+
+if __name__ == '__main__':
+    unittest.main()