summaryrefslogtreecommitdiff
path: root/numpy/lib
diff options
context:
space:
mode:
authorKenichi Maehashi <webmaster@kenichimaehashi.com>2021-06-10 12:02:15 +0000
committerKenichi Maehashi <webmaster@kenichimaehashi.com>2021-06-10 12:03:25 +0000
commit4dead884769118cd9af8bd5210273291804429e7 (patch)
tree677fdbafd2bcd029b4a0f47f7e6f4139c746fb90 /numpy/lib
parent04ab04d93d4d7a4d241fe0ceb725436a8b6c8c2e (diff)
downloadnumpy-4dead884769118cd9af8bd5210273291804429e7.tar.gz
ENH: support major version larger than 9 in numpy.lib.NumpyVersion
Diffstat (limited to 'numpy/lib')
-rw-r--r--numpy/lib/_version.py2
-rw-r--r--numpy/lib/tests/test__version.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/numpy/lib/_version.py b/numpy/lib/_version.py
index 00e00e9a7..145c54855 100644
--- a/numpy/lib/_version.py
+++ b/numpy/lib/_version.py
@@ -54,7 +54,7 @@ class NumpyVersion():
def __init__(self, vstring):
self.vstring = vstring
- ver_main = re.match(r'\d\.\d+\.\d+', vstring)
+ ver_main = re.match(r'\d+\.\d+\.\d+', vstring)
if not ver_main:
raise ValueError("Not a valid numpy version string")
diff --git a/numpy/lib/tests/test__version.py b/numpy/lib/tests/test__version.py
index 182504631..e6d41ad93 100644
--- a/numpy/lib/tests/test__version.py
+++ b/numpy/lib/tests/test__version.py
@@ -7,7 +7,7 @@ from numpy.lib import NumpyVersion
def test_main_versions():
assert_(NumpyVersion('1.8.0') == '1.8.0')
- for ver in ['1.9.0', '2.0.0', '1.8.1']:
+ for ver in ['1.9.0', '2.0.0', '1.8.1', '10.0.1']:
assert_(NumpyVersion('1.8.0') < ver)
for ver in ['1.7.0', '1.7.1', '0.9.9']: