def test_2():
"""
basic 3-d unit test
"""
n1 = 10
n2 = 100
ndim = 3
semi_axes = np.random.random((n1, ndim))
coords = np.array([sample_ellipsoidal_volume(n2, semi_axes[i]) for i in range(0, n1)])
Is = reduced_inertia_tensors(coords)
assert np.shape(Is) == (n1, ndim, ndim)
def test_1():
"""
basic 2-d unit test
"""
n1 = 10
n2 = 100
ndim = 2
semi_axes = np.random.random((n1, ndim))
coords = [
sample_ellipsoidal_volume(n2, semi_axes[i]) for i in range(0, n1)
]
Is = inertia_tensors(coords)
assert np.shape(Is) == (n1, ndim, ndim)
def test_3():
"""
test weights argument
"""
n1 = 10
n2 = 100
ndim = 3
semi_axes = np.random.random((n1, ndim))
coords = np.array([sample_ellipsoidal_volume(n2, semi_axes[i]) for i in range(0, n1)])
weights = np.ones((n1, n2))
Is_1 = reduced_inertia_tensors(coords, weights=weights)
Is_2 = reduced_inertia_tensors(coords, weights=None)
assert np.shape(Is_1) == (n1, ndim, ndim)
assert np.shape(Is_2) == (n1, ndim, ndim)
assert np.allclose(Is_1, Is_2)
