TST: Add typing tests for `np.lib.NumpyVersion`

3 files changed, 42 insertions, 0 deletions

diff --git a/numpy/typing/tests/data/fail/lib_version.py b/numpy/typing/tests/data/fail/lib_version.py
new file mode 100644
index 000000000..0769068d4
--- /dev/null
+++ b/numpy/typing/tests/data/fail/lib_version.py
@@ -0,0 +1,6 @@
+from numpy.lib import NumpyVersion
+
+version: NumpyVersion
+
+NumpyVersion(b"1.8.0")  # E: incompatible type
+version >= b"1.8.0"  # E: incompatible type
diff --git a/numpy/typing/tests/data/pass/lib_version.py b/numpy/typing/tests/data/pass/lib_version.py
new file mode 100644
index 000000000..f3825eca5
--- /dev/null
+++ b/numpy/typing/tests/data/pass/lib_version.py
@@ -0,0 +1,18 @@
+from numpy.lib import NumpyVersion
+
+version = NumpyVersion("1.8.0")
+
+version.vstring
+version.version
+version.major
+version.minor
+version.bugfix
+version.pre_release
+version.is_devversion
+
+version == version
+version != version
+version < "1.8.0"
+version <= version
+version > version
+version >= "1.8.0"
diff --git a/numpy/typing/tests/data/reveal/lib_version.py b/numpy/typing/tests/data/reveal/lib_version.py
new file mode 100644
index 000000000..e6f695558
--- /dev/null
+++ b/numpy/typing/tests/data/reveal/lib_version.py
@@ -0,0 +1,18 @@
+from numpy.lib import NumpyVersion
+
+version = NumpyVersion("1.8.0")
+
+reveal_type(version.vstring)  # E: str
+reveal_type(version.version)  # E: str
+reveal_type(version.major)  # E: int
+reveal_type(version.minor)  # E: int
+reveal_type(version.bugfix)  # E: int
+reveal_type(version.pre_release)  # E: str
+reveal_type(version.is_devversion)  # E: bool
+
+reveal_type(version == version)  # E: bool
+reveal_type(version != version)  # E: bool
+reveal_type(version < "1.8.0")  # E: bool
+reveal_type(version <= version)  # E: bool
+reveal_type(version > version)  # E: bool
+reveal_type(version >= "1.8.0")  # E: bool