tx258/lammps-gran-kokkos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix issues causing failed tests

Browse Source
This commit is contained in:
Axel Kohlmeyer
2025-06-18 07:31:10 -04:00
parent 004816a4ba
commit 8c0894fa84
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/lammps/core.py
View File

@ -87,9 +87,9 @@ class command_wrapper:
args = list(args) args = list(args)
args[0] = lammps(ptr=args[0]) args[0] = lammps(ptr=args[0])
x(*args) x(*args)
setattr(__main__, func_name, handler) setattr(__main__, func_name, handler)
return func_name return func_name
return x return x
def __getattr__(self, name): def __getattr__(self, name):
""" """
@ -112,7 +112,7 @@ class command_wrapper:
# Python 3.6+ maintains ordering of kwarg keys # Python 3.6+ maintains ordering of kwarg keys
for kw, arg in kwargs.items(): for kw, arg in kwargs.items():
cmd_args.append(kw) cmd_args.append(kw)
if isinstance(arg,bool): if isinstance(arg, bool):
cmd_args.append("true" if arg else "false") cmd_args.append("true" if arg else "false")
else: else:
cmd_args.append(str(self._wrap_args(arg))) cmd_args.append(str(self._wrap_args(arg)))

2
unittest/python/python-cmdwrapper.py
View File

@ -53,7 +53,7 @@ class PythonCmdWrapper(unittest.TestCase):
types = [1, 1] types = [1, 1]
self.assertEqual(self.lmp.create_atoms(2, id=None, type=types, x=x), 2) self.assertEqual(self.lmp.create_atoms(2, atomid=None, atype=types, x=x), 2)
self.assertEqual(self.lmp.extract_global("natoms"), 2) self.assertEqual(self.lmp.extract_global("natoms"), 2)
pos = self.lmp.numpy.extract_atom("x") pos = self.lmp.numpy.extract_atom("x")
self.assertEqual(pos.shape[0], 2) self.assertEqual(pos.shape[0], 2)