def test_geometric_objects_duplicates(self):
rad = 1
s = mp.Sphere(rad)
c = mp.Cylinder(rad)
res = mp.geometric_objects_duplicates(mp.Vector3(1, 1, 1), 1, 5, [s, c])
expected = [
mp.Sphere(rad, center=mp.Vector3(5, 5, 5)),
mp.Sphere(rad, center=mp.Vector3(4, 4, 4)),
mp.Sphere(rad, center=mp.Vector3(3, 3, 3)),
mp.Sphere(rad, center=mp.Vector3(2, 2, 2)),
mp.Sphere(rad, center=mp.Vector3(1, 1, 1)),
mp.Cylinder(rad, center=mp.Vector3(5, 5, 5)),
mp.Cylinder(rad, center=mp.Vector3(4, 4, 4)),
mp.Cylinder(rad, center=mp.Vector3(3, 3, 3)),
mp.Cylinder(rad, center=mp.Vector3(2, 2, 2)),
mp.Cylinder(rad, center=mp.Vector3(1, 1, 1)),
]
for r, e in zip(res, expected):
self.assertEqual(r.center, e.center)
def test_geometric_objects_duplicates(self):
rad = 1
s = mp.Sphere(rad)
c = mp.Cylinder(rad)
res = mp.geometric_objects_duplicates(mp.Vector3(1, 1, 1), 1, 5, [s, c])
expected = [
mp.Sphere(rad, center=mp.Vector3(5, 5, 5)),
mp.Sphere(rad, center=mp.Vector3(4, 4, 4)),
mp.Sphere(rad, center=mp.Vector3(3, 3, 3)),
mp.Sphere(rad, center=mp.Vector3(2, 2, 2)),
mp.Sphere(rad, center=mp.Vector3(1, 1, 1)),
mp.Cylinder(rad, center=mp.Vector3(5, 5, 5)),
mp.Cylinder(rad, center=mp.Vector3(4, 4, 4)),
mp.Cylinder(rad, center=mp.Vector3(3, 3, 3)),
mp.Cylinder(rad, center=mp.Vector3(2, 2, 2)),
mp.Cylinder(rad, center=mp.Vector3(1, 1, 1)),
]
for r, e in zip(res, expected):
self.assertEqual(r.center, e.center)