Ejemplo n.º 1
0
    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())
Ejemplo n.º 2
0
 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())