switch from xp.asarray(list(range to nx.arange

Author: Enderdead

tests/test_funcs.py

@@ -311,7 +311,7 @@ def test_1D_shapes(
         xp: ModuleType,
     ):
         n = math.prod(input_shape)
-        x = xp.reshape(xp.asarray(list(range(n))), input_shape)
+        x = xp.reshape(xp.arange(n), input_shape)
         y = atleast_nd(x, ndim=ndim)
 
         assert y.shape == expected_shape
@@ -350,7 +350,7 @@ def test_2D_shapes(
         xp: ModuleType,
     ):
         n = math.prod(input_shape)
-        x = xp.reshape(xp.asarray(list(range(n))), input_shape)
+        x = xp.reshape(xp.arange(n), input_shape)
         y = atleast_nd(x, ndim=ndim)
 
         assert y.shape == expected_shape
@@ -389,7 +389,7 @@ def test_3D_shapes(
         xp: ModuleType,
     ):
         n = math.prod(input_shape)
-        x = xp.reshape(xp.asarray(list(range(n))), input_shape)
+        x = xp.reshape(xp.arange(n), input_shape)
         y = atleast_nd(x, ndim=ndim)
 
         assert y.shape == expected_shape
@@ -428,7 +428,7 @@ def test_5D_shapes(
         xp: ModuleType,
     ):
         n = math.prod(input_shape)
-        x = xp.reshape(xp.asarray(list(range(n))), input_shape)
+        x = xp.reshape(xp.arange(n), input_shape)
         y = atleast_nd(x, ndim=ndim)
 
         assert y.shape == expected_shape