From 55feca9c0fb4a82ce721b009760e9b5534ae8601 Mon Sep 17 00:00:00 2001 From: Axel Kohlmeyer Date: Sun, 21 Aug 2022 14:02:06 -0400 Subject: [PATCH] add minimal tests for dump class --- unittest/python/python-pizza.py | 28 ++++++++++++++++++++++++++-- 1 file changed, 26 insertions(+), 2 deletions(-) diff --git a/unittest/python/python-pizza.py b/unittest/python/python-pizza.py index 5773b0c430..cf0aff2e5a 100644 --- a/unittest/python/python-pizza.py +++ b/unittest/python/python-pizza.py @@ -43,7 +43,7 @@ class Logfiles(unittest.TestCase): self.assertEqual(m[2],0.0) l.write("all.txt",0) l.write("some.txt",1,"Step","Temp","Press") - + with self.assertRaises(Exception): t = l.next() @@ -58,7 +58,7 @@ class Logfiles(unittest.TestCase): t = l.next() self.assertEqual(l.style, 2) self.assertEqual(l.nvec, 6) - + def testMultiLogFile(self): l = log.log(os.path.join(EXAMPLES_DIR, MULTI_STYLE_EXAMPLE_LOG)) self.assertEqual(l.nvec, 14) @@ -120,6 +120,30 @@ class PythonDump(unittest.TestCase): self.lmp.command("dump 1 all custom 2 " + dumpfile + " id type mol q x y z vx vy vz") self.lmp.command("dump_modify 1 time yes units yes") self.lmp.command("run 4 post no") + d = dump.dump(dumpfile) + id1, id2 = d.minmax("id") + self.assertEqual(id1,1) + self.assertEqual(id2,29) + t = d.time() + self.assertEqual(len(t),3) + d.tselect.one(2,4) + index, time, flag = d.iterator(0) + self.assertEqual(index,1) + self.assertEqual(time,2) + self.assertEqual(flag,1) + index, time, flag = d.iterator(1) + self.assertEqual(index,2) + self.assertEqual(time,4) + self.assertEqual(flag,1) + index, time, flag = d.iterator(1) + self.assertEqual(index,0) + self.assertEqual(time,0) + self.assertEqual(flag,-1) + + with self.assertRaises(Exception): + t = d.next() + + os.remove(dumpfile) if __name__ == "__main__": unittest.main()