#NEWLINE# def test_rechunk_expand( ) : #NEWLINE# #INDENT# a = np . random . uniform ( 0 , 1 , 100 ) . reshape ( ( 10 , 10 ) ) #NEWLINE# x = da . from_array ( a , chunks = ( 5 , 5 ) ) #NEWLINE# y = x . rechunk ( chunks = ( ( 3 , 3 , 3 , 1 ) , ( 3 , 3 , 3 , 1 ) ) ) #NEWLINE# assert np . all ( ( y . compute ( ) == a ) )