diff --git a/unittest/python/python-numpy.py b/unittest/python/python-numpy.py index 67dee454a1..9be5ddbaf7 100644 --- a/unittest/python/python-numpy.py +++ b/unittest/python/python-numpy.py @@ -1,5 +1,5 @@ import sys,os,unittest -from lammps import lammps, LMP_STYLE_GLOBAL, LMP_STYLE_ATOM, LMP_TYPE_VECTOR, LMP_TYPE_SCALAR +from lammps import lammps, LMP_STYLE_GLOBAL, LMP_STYLE_LOCAL, LMP_STYLE_ATOM, LMP_TYPE_VECTOR, LMP_TYPE_SCALAR, LMP_TYPE_ARRAY from ctypes import c_void_p class PythonNumpy(unittest.TestCase): @@ -15,7 +15,10 @@ class PythonNumpy(unittest.TestCase): def testLammpsPointer(self): self.assertEqual(type(self.lmp.lmp), c_void_p) - def testExtractCompute(self): + def testExtractComputeGlobalScalar(self): + # TODO + + def testExtractComputeGlobalVector(self): self.lmp.command("region box block 0 2 0 2 0 2") self.lmp.command("create_box 1 box") self.lmp.command("create_atoms 1 single 1.0 1.0 1.0") @@ -29,7 +32,10 @@ class PythonNumpy(unittest.TestCase): self.assertEqual(values[1], 2.0) self.assertEqual(values[2], 2.5) - def testExtractComputePerAtom(self): + def testExtractComputeGlobalArray(self): + # TODO + + def testExtractComputePerAtomVector(self): self.lmp.command("region box block 0 2 0 2 0 2") self.lmp.command("create_box 1 box") self.lmp.command("create_atoms 1 single 1.0 1.0 1.0") @@ -42,5 +48,17 @@ class PythonNumpy(unittest.TestCase): self.assertEqual(values[0], 0.0) self.assertEqual(values[1], 0.0) + def testExtractComputePerAtomArray(self): + # TODO + + def testExtractComputeLocalScalar(self): + # TODO + + def testExtractComputeLocalVector(self): + # TODO + + def testExtractComputeLocalArray(self): + # TODO + if __name__ == "__main__": unittest.main()