Add extra tests

openkim · Mar 13, 2024 · 696eb29 · 696eb29
1 parent abebcd3
commit 696eb29
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/tests/test_kim_edn/test_dump.py b/tests/test_kim_edn/test_dump.py
@@ -1,4 +1,5 @@
 from io import StringIO
+import os
 from tests.test_kim_edn import PyTest
 
 
@@ -9,6 +10,14 @@ def test_dump(self):
         self.kim_edn.dump({}, sio)
         self.assertEqual(sio.getvalue(), '{}\n')   # NOTE
 
+        d = {1337: "true.ed"}
+        self.addCleanup(os.remove, d[1337])
+
+        self.kim_edn.dump(d, d[1337])
+
+        self.assertEqual(self.kim_edn.load('{"1337" "true.ed"}'),
+                         self.kim_edn.load(d[1337]))
+
     def test_dumps(self):
         self.assertEqual(self.dumps({}), '{}')
 
@@ -36,10 +45,10 @@ def __lt__(self, o):
 
         L = [X() for i in range(1122)]
         d = D()
-        d[hash(d.keys()[0])] = "true.dat"
+        d[hash(d.keys()[0])] = "true.edn"
 
         self.assertEqual(self.dumps(d, sort_keys=(not L[0] < L[1])),
-                         '{"1337" "true.dat"}')
+                         '{"1337" "true.edn"}')
 
 
 class TestPyDump(TestDump, PyTest):