np.testing.assert_allclose(aret[0].astype(float),
                                       flats.reshape(arr_shape).astype(float))