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))