def test_mosaic():
a = seq_array((5,4,3,2))
shape = [5, 24]
assert_array_equal(shape, mosaic(a).shape)
def test_mosaic_masked():
a = seq_array((5,4,3,2))
x = np.ma.array(a, mask=(a % 2 == 0))
shape = [5, 24]
assert_array_equal(shape, mosaic(x).shape)
assert_array_equal(shape, mosaic(x).mask.shape)
def test_mosaic_2d():
a = seq_array((5,4))
assert_array_equal(a, mosaic(a))
