summaryrefslogtreecommitdiff
path: root/numpy/tests
diff options
context:
space:
mode:
authorBas van Beek <b.f.van.beek@vu.nl>2020-08-23 17:59:35 +0200
committerBas van Beek <b.f.van.beek@vu.nl>2020-08-23 17:59:35 +0200
commitcabee5ecffe634d8f9aebe7f30427cbb1052aeb7 (patch)
tree706b91074dee467e11536aed70c0d0d42f5d22e7 /numpy/tests
parent72e11d38e3d8acadd1d5341e1c0b4b17d0eda6cb (diff)
downloadnumpy-cabee5ecffe634d8f9aebe7f30427cbb1052aeb7.tar.gz
MAINT: The returned step size can be complex
Co-Authored-By: Eric Wieser <eric-wieser@users.noreply.github.com>
Diffstat (limited to 'numpy/tests')
-rw-r--r--numpy/tests/typing/pass/linspace.py1
-rw-r--r--numpy/tests/typing/reveal/linspace.py2
2 files changed, 2 insertions, 1 deletions
diff --git a/numpy/tests/typing/pass/linspace.py b/numpy/tests/typing/pass/linspace.py
index ce427e866..c75f5bf66 100644
--- a/numpy/tests/typing/pass/linspace.py
+++ b/numpy/tests/typing/pass/linspace.py
@@ -7,6 +7,7 @@ np.linspace(0j, 2)
np.linspace(0, 2, num=10)
np.linspace(0, 2, endpoint=True)
np.linspace(0, 2, retstep=True)
+np.linspace(0j, 2j, retstep=True)
np.linspace(0, 2, dtype=bool)
np.linspace([0, 1], [2, 3], axis=1)
diff --git a/numpy/tests/typing/reveal/linspace.py b/numpy/tests/typing/reveal/linspace.py
index 678a5d3bf..cfbbdf390 100644
--- a/numpy/tests/typing/reveal/linspace.py
+++ b/numpy/tests/typing/reveal/linspace.py
@@ -1,6 +1,6 @@
import numpy as np
reveal_type(np.linspace(0, 10)) # E: numpy.ndarray
-reveal_type(np.linspace(0, 10, retstep=True)) # E: Tuple[numpy.ndarray, numpy.floating]
+reveal_type(np.linspace(0, 10, retstep=True)) # E: Tuple[numpy.ndarray, numpy.inexact]
reveal_type(np.logspace(0, 10)) # E: numpy.ndarray
reveal_type(np.geomspace(1, 10)) # E: numpy.ndarray