Skip to content

Commit e586075

Browse files
Add assert on USM data pointer to tests
1 parent 82de136 commit e586075

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

dpnp/tests/test_linalg.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1931,6 +1931,7 @@ def test_overwrite_inplace(self, dtype):
19311931
)
19321932

19331933
assert lu is a_dp
1934+
assert lu.data.ptr == a_dp.data.ptr
19341935
assert lu.flags["F_CONTIGUOUS"] is True
19351936

19361937
L, U = self._split_lu(lu, 2, 2)
@@ -1948,6 +1949,7 @@ def test_overwrite_copy(self, dtype):
19481949
)
19491950

19501951
assert lu is not a_dp
1952+
assert lu.data.ptr != a_dp.data.ptr
19511953
assert lu.flags["F_CONTIGUOUS"] is True
19521954

19531955
L, U = self._split_lu(lu, 2, 2)
@@ -1974,6 +1976,7 @@ def test_overwrite_copy_special(self):
19741976
)
19751977

19761978
assert lu is not a_dp
1979+
assert lu.data.ptr != a_dp.data.ptr
19771980
assert lu.flags["F_CONTIGUOUS"] is True
19781981

19791982
L, U = self._split_lu(lu, 2, 2)
@@ -2217,6 +2220,7 @@ def test_overwrite_inplace(self, dtype):
22172220
)
22182221

22192222
assert x is b_dp
2223+
assert x.data.ptr == b_dp.data.ptr
22202224
assert dpnp.allclose(a_dp @ x, b_orig, rtol=1e-6, atol=1e-6)
22212225

22222226
@pytest.mark.parametrize("dtype", get_float_complex_dtypes())
@@ -2230,6 +2234,7 @@ def test_overwrite_copy_special(self, dtype):
22302234
(lu, piv), b1, overwrite_b=True, check_finite=False
22312235
)
22322236
assert x1 is not b1
2237+
assert x1.data.ptr != b1.data.ptr
22332238

22342239
# F-contig, match dtype but read-only input
22352240
b2 = dpnp.array([1, 0], dtype=dtype, order="F")
@@ -2238,6 +2243,7 @@ def test_overwrite_copy_special(self, dtype):
22382243
(lu, piv), b2, overwrite_b=True, check_finite=False
22392244
)
22402245
assert x2 is not b2
2246+
assert x2.data.ptr != b2.data.ptr
22412247

22422248
for x in (x1, x2):
22432249
assert dpnp.allclose(

0 commit comments

Comments
 (0)