def test_to_numpy(self): vals = np.random.rand(8, 8) row_arr = RowMajorArray(vals) np.testing.assert_almost_equal(vals, row_arr.to_numpy()) vals = np.random.rand(8, 8, 8) row_arr = RowMajorArray(vals) np.testing.assert_almost_equal(vals, row_arr.to_numpy())
def test_setting_to_3d_pic(self): vals = np.random.rand(4, 4, 4) row_arr = RowMajorArray(picture=vals) assert row_arr.dim == 3 assert row_arr.shape == vals.shape np.testing.assert_almost_equal(vals, row_arr.to_numpy())