def test_many_shapes():
"""Test many shapes."""
assert number_of_shapes(np.random.random((8, 4, 2))) == 8
def test_no_shapes():
"""Test no shapes."""
assert number_of_shapes([]) == 0
assert number_of_shapes(np.empty((0, 4, 2))) == 0
def test_one_shape():
"""Test one shape."""
assert number_of_shapes(np.random.random((4, 2))) == 1