def test_coord_wrap_box(self):
xyz = np.array([[3, 3, 1], [1, 1, 0]])
box = mb.Box(lengths=[2.0, 2.0, 2.0], angles=[90, 90, 90])
new_xyz = wrap_coords(xyz, box, mins=np.min(xyz, axis=0))
assert (new_xyz[0, :] == np.array([1, 1, 1])).all()
assert (new_xyz[1, :] == xyz[1, :]).all()
def test_coord_wrap_box(self):
xyz = np.array([[3, 3, 1], [1, 1, 0]])
box = mb.Box(mins=[-2.0, -2.0, -2.0], maxs=[2.0, 2.0, 2.0])
new_xyz = wrap_coords(xyz, box)
assert (new_xyz[0, :] == np.array([-1, -1, 1])).all()
assert (new_xyz[1, :] == xyz[1, :]).all()
def test_coord_wrap(self):
xyz = np.array([[3, 3, 1], [1, 1, 0]])
box = [2, 2, 2]
new_xyz = wrap_coords(xyz, box)
assert (new_xyz[0, :] == np.array([1, 1, 1])).all()
assert (new_xyz[1, :] == xyz[1, :]).all()
