Skip to content

Commit 7d1fd0b

Browse files
Update data inputs for test_usm_type
1 parent e586075 commit 7d1fd0b

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

dpnp/tests/test_sycl_queue.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1612,13 +1612,13 @@ def test_lu_factor(self, data, device):
16121612
assert_sycl_queue_equal(param_queue, a.sycl_queue)
16131613

16141614
@pytest.mark.parametrize(
1615-
"data",
1615+
"b_data",
16161616
[[1.0, 2.0], numpy.empty((2, 0))],
16171617
)
1618-
def test_lu_solve(self, data, device):
1618+
def test_lu_solve(self, b_data, device):
16191619
a = dpnp.array([[1.0, 2.0], [3.0, 5.0]], device=device)
16201620
lu, piv = dpnp.linalg.lu_factor(a)
1621-
b = dpnp.array(data, device=device)
1621+
b = dpnp.array(b_data, device=device)
16221622

16231623
result = dpnp.linalg.lu_solve((lu, piv), b)
16241624

dpnp/tests/test_usm_type.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1489,13 +1489,13 @@ def test_lu_factor(self, data, usm_type):
14891489

14901490
@pytest.mark.parametrize("usm_type_rhs", list_of_usm_types)
14911491
@pytest.mark.parametrize(
1492-
"data",
1492+
"b_data",
14931493
[[1.0, 2.0], numpy.empty((2, 0))],
14941494
)
1495-
def test_lu_solve(self, data, usm_type, usm_type_rhs):
1496-
a = dpnp.array(data, usm_type=usm_type)
1495+
def test_lu_solve(self, b_data, usm_type, usm_type_rhs):
1496+
a = dpnp.array([[1.0, 2.0], [3.0, 5.0]], usm_type=usm_type)
14971497
lu, piv = dpnp.linalg.lu_factor(a)
1498-
b = dpnp.array(data, usm_type=usm_type_rhs)
1498+
b = dpnp.array(b_data, usm_type=usm_type_rhs)
14991499

15001500
result = dpnp.linalg.lu_solve((lu, piv), b)
15011501

0 commit comments

Comments
 (0)