Adding mass to python numpy unittest of KE

This commit is contained in:
jtclemm
2023-03-27 16:37:11 -06:00
committed by Axel Kohlmeyer
parent 824ed3379a
commit 7738b67737

View File

@ -108,6 +108,7 @@ class PythonNumpy(unittest.TestCase):
self.lmp.command("create_box 1 box") self.lmp.command("create_box 1 box")
self.lmp.command("create_atoms 1 single 1.0 1.0 1.0") self.lmp.command("create_atoms 1 single 1.0 1.0 1.0")
self.lmp.command("create_atoms 1 single 1.0 1.0 1.5") self.lmp.command("create_atoms 1 single 1.0 1.0 1.5")
self.lmp.command("mass 1 1.0")
self.lmp.command("compute ke all ke/atom") self.lmp.command("compute ke all ke/atom")
natoms = self.lmp.get_natoms() natoms = self.lmp.get_natoms()
self.assertEqual(natoms,2) self.assertEqual(natoms,2)