Update linspace tests

antonwolfy · antonwolfy · commit 52b8052752fb · 2025-01-19T12:50:27.000+01:00
diff --git a/dpnp/tests/test_arraycreation.py b/dpnp/tests/test_arraycreation.py
@@ -17,6 +17,7 @@
 from .helper import (
     assert_dtype_allclose,
     get_all_dtypes,
+    get_array,
 )
 from .third_party.cupy import testing
 
@@ -768,7 +769,7 @@ def test_space_numpy_dtype(func, start_dtype, stop_dtype):
     ],
 )
 def test_linspace_arrays(start, stop):
-    func = lambda xp: xp.linspace(start, stop, 10)
+    func = lambda xp: xp.linspace(get_array(xp, start), get_array(xp, stop), 10)
     assert func(numpy).shape == func(dpnp).shape
 
 

Original file line number	Diff line number	Diff line change
`@@ -17,6 +17,7 @@`
`17`	`17`	`from .helper import (`
`18`	`18`	`assert_dtype_allclose,`
`19`	`19`	`get_all_dtypes,`
	`20`	`+ get_array,`
`20`	`21`	`)`
`21`	`22`	`from .third_party.cupy import testing`
`22`	`23`
`@@ -768,7 +769,7 @@ def test_space_numpy_dtype(func, start_dtype, stop_dtype):`
`768`	`769`	`],`
`769`	`770`	`)`
`770`	`771`	`def test_linspace_arrays(start, stop):`
`771`		`- func = lambda xp: xp.linspace(start, stop, 10)`
	`772`	`+ func = lambda xp: xp.linspace(get_array(xp, start), get_array(xp, stop), 10)`
`772`	`773`	`assert func(numpy).shape == func(dpnp).shape`
`773`	`774`
`774`	`775`