def test_cpso_remove():
c = 3.
n = 4
cpso = CPSO("foo")
objs = [RBSO(str(i)) for i in xrange(n)]
for obj in objs:
obj.setPos((c * (i - np.floor(n / 2.)), 0, 0))
obj.set_shape("Box")
cpso.add(objs)
assert len(cpso.getChildren()) == n
cpso.remove(objs[:-2])
assert len(cpso.getChildren()) == 2
def test_cpso_add():
c = 3.
n = 3
cpso = CPSO("foo")
objs = [RBSO(str(i)) for i in xrange(n)]
for obj in objs:
obj.setPos((c * i - c, 0, 0))
obj.set_shape("Box")
cpso.add(objs)
assert len(cpso.getChildren()) == n
