summaryrefslogtreecommitdiff
path: root/numpy/tests/test_numpy_version.py
diff options
context:
space:
mode:
authorCharles Harris <charlesr.harris@gmail.com>2021-05-31 13:43:20 -0600
committerGitHub <noreply@github.com>2021-05-31 13:43:20 -0600
commit9c0809f7799c8b0297786c2df59cba622a7a9dab (patch)
treedd0dfd8d7a74529472985da86442939945461b64 /numpy/tests/test_numpy_version.py
parent189110e8df94b621d4b2cf1d1b8453fd7dd2b949 (diff)
parentabf9fe9add589166c383e2cc0674499683fcaa7c (diff)
downloadnumpy-9c0809f7799c8b0297786c2df59cba622a7a9dab.tar.gz
Merge pull request #19142 from kif/short_version
BUG: expose `short_version` as previously in version.py
Diffstat (limited to 'numpy/tests/test_numpy_version.py')
-rw-r--r--numpy/tests/test_numpy_version.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/numpy/tests/test_numpy_version.py b/numpy/tests/test_numpy_version.py
index 28595026e..bccbcb8e9 100644
--- a/numpy/tests/test_numpy_version.py
+++ b/numpy/tests/test_numpy_version.py
@@ -32,3 +32,13 @@ def test_valid_numpy_version():
res = re.match(version_pattern + dev_suffix + '$', np.__version__)
assert_(res is not None, np.__version__)
+
+
+def test_short_version():
+ # Check numpy.short_version actually exists
+ if np.version.release:
+ assert_(np.__version__ == np.version.short_version,
+ "short_version mismatch in release version")
+ else:
+ assert_(np.__version__.split("+")[0] == np.version.short_version,
+ "short_version mismatch in development version")