add test for axis in logspace
@@ -284,6 +284,14 @@ class TestRanges(unittest.TestCase):
def test_logspace_base(self, xp, dtype):
return xp.logspace(0, 2, 5, base=2.0, dtype=dtype)
+ @testing.with_requires('numpy>=1.16')
+ @testing.for_all_dtypes_combination(names=('dtype_range', 'dtype_out'),
+ no_bool=True, no_complex=True)
+ @testing.numpy_cupy_array_equal()
+ def test_logspace_array_start_stop_axis1(self, xp, dtype_range, dtype_out):
+ start = xp.array([0, 2], dtype=dtype_range)
+ stop = xp.array([2, 0], dtype=dtype_range)
+ return xp.logspace(start, stop, num=5, dtype=dtype_out, axis=1)
@testing.parameterize(
*testing.product({
