diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2020-11-03 09:20:31 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-11-03 09:20:31 -0700 |
| commit | 8b15e57718042c75af22a25a7d604fa0f938f16e (patch) | |
| tree | 0595ce56ac5659b47faef356664d012690ac73ec /numpy/typing | |
| parent | d62b0ee88b20e5946fe49f0ba533b3e547e4d4f1 (diff) | |
| parent | 66b17fbc9666b150c71bb94f2492fd880b2641e4 (diff) | |
| download | numpy-8b15e57718042c75af22a25a7d604fa0f938f16e.tar.gz | |
Merge pull request #17701 from BvB93/pyi
BUG: Fixed an issue where `.pyi` files were ignored by numpy sub-packages
Diffstat (limited to 'numpy/typing')
| -rw-r--r-- | numpy/typing/setup.py | 1 | ||||
| -rw-r--r-- | numpy/typing/tests/test_isfile.py | 34 |
2 files changed, 35 insertions, 0 deletions
diff --git a/numpy/typing/setup.py b/numpy/typing/setup.py index c444e769f..694a756dc 100644 --- a/numpy/typing/setup.py +++ b/numpy/typing/setup.py @@ -3,6 +3,7 @@ def configuration(parent_package='', top_path=None): config = Configuration('typing', parent_package, top_path) config.add_subpackage('tests') config.add_data_dir('tests/data') + config.add_data_files('*.pyi') return config diff --git a/numpy/typing/tests/test_isfile.py b/numpy/typing/tests/test_isfile.py new file mode 100644 index 000000000..e6b80f995 --- /dev/null +++ b/numpy/typing/tests/test_isfile.py @@ -0,0 +1,34 @@ +import os +from pathlib import Path + +import numpy as np +from numpy.testing import assert_ + +ROOT = Path(np.__file__).parents[0] +FILES = [ + ROOT / "py.typed", + ROOT / "__init__.pyi", + ROOT / "char.pyi", + ROOT / "ctypeslib.pyi", + ROOT / "emath.pyi", + ROOT / "rec.pyi", + ROOT / "version.pyi", + ROOT / "core" / "__init__.pyi", + ROOT / "distutils" / "__init__.pyi", + ROOT / "f2py" / "__init__.pyi", + ROOT / "fft" / "__init__.pyi", + ROOT / "lib" / "__init__.pyi", + ROOT / "linalg" / "__init__.pyi", + ROOT / "ma" / "__init__.pyi", + ROOT / "matrixlib" / "__init__.pyi", + ROOT / "polynomial" / "__init__.pyi", + ROOT / "random" / "__init__.pyi", + ROOT / "testing" / "__init__.pyi", +] + + +class TestIsFile: + def test_isfile(self): + """Test if all ``.pyi`` files are properly installed.""" + for file in FILES: + assert_(os.path.isfile(file)) |