add tests

pyiron · Nov 20, 2024 · 4dd1e87 · 4dd1e87
1 parent cbfecf6
commit 4dd1e87
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/tests/unit/test_input.py b/tests/unit/test_input.py
@@ -0,0 +1,22 @@
+import unittest
+from stinx.input import sphinx
+
+
+class TestStinx(unittest.TestCase):
+
+    def test_wrap_string(self):
+        with_wrap_string = sphinx.pawPot.species.create(
+            potential="my_potential_path",
+            potType="AtomPAW",
+        )
+        self.assertEqual(with_wrap_string, {'potential': '"my_potential_path"', 'potType': '"AtomPAW"'})
+        without_wrap_string = sphinx.pawPot.species.create(
+            potential="my_potential_path",
+            potType="AtomPAW",
+            wrap_string=False,
+        )
+        self.assertEqual(without_wrap_string, {'potential': 'my_potential_path', 'potType': 'AtomPAW'})
+
+
+if __name__ == "__main__":
+    unittest.main()