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