Int32 is too small for geomspace test

Author: ricardoV94

tests/tensor/test_extra_ops.py

@@ -1301,11 +1301,13 @@ def test_broadcast_arrays():
     ["linspace", "logspace", "geomspace"],
     ids=["linspace", "logspace", "geomspace"],
 )
-@pytest.mark.parametrize("dtype", [None, "int", "float"], ids=[None, "int", "float"])
+@pytest.mark.parametrize(
+    "dtype", [None, "int64", "floatX"], ids=[None, "int64", "floatX"]
+)
 @pytest.mark.parametrize(
     "start, stop, num_samples, endpoint, axis",
     [
-        (1, 10, 50, True, 0),
+        (1, 10, 40, True, 0),
         (1, 10, 1, True, 0),
         (np.array([5, 6]), np.array([[10, 10], [10, 10]]), 25, True, 0),
         (np.array([5, 6]), np.array([[10, 10], [10, 10]]), 25, True, 1),
@@ -1317,7 +1319,7 @@ def test_broadcast_arrays():
 def test_space_ops(op, dtype, start, stop, num_samples, endpoint, axis):
     pt_func = getattr(pt, op)
     np_func = getattr(np, op)
-    dtype = dtype + config.floatX[-2:] if dtype is not None else dtype
+    dtype = dtype if dtype != "floatX" else config.floatX
     z = pt_func(start, stop, num_samples, endpoint=endpoint, axis=axis, dtype=dtype)
 
     numpy_res = np_func(