Add unit test for extract_variable (equalstyle)

2020-11-23 13:07:00 -05:00
parent 875057538f
commit 5ff0c3d4f0
1 changed files with 8 additions and 0 deletions
--- a/unittest/python/python-commands.py
+++ b/unittest/python/python-commands.py
@ -192,6 +192,14 @@ create_atoms 1 single &
        self.assertEqual(periodicity, [1, 1, 1])
        self.assertEqual(box_change, 0)

+    def test_extract_variable_equalstyle(self):
+        self.lmp.command("variable a equal 100")
+        a = self.lmp.extract_variable("a")
+        self.assertEqual(a, 100)
+
+        self.lmp.command("variable a equal 3.14")
+        a = self.lmp.extract_variable("a")
+        self.assertEqual(a, 3.14)

 ##############################
 if __name__ == "__main__":