test_isfile.py 865 B

1234567891011121314151617181920212223242526272829303132
  1. import os
  2. import sys
  3. from pathlib import Path
  4. import numpy as np
  5. from numpy.testing import assert_
  6. ROOT = Path(np.__file__).parents[0]
  7. FILES = [
  8. ROOT / "py.typed",
  9. ROOT / "__init__.pyi",
  10. ROOT / "ctypeslib.pyi",
  11. ROOT / "_core" / "__init__.pyi",
  12. ROOT / "f2py" / "__init__.pyi",
  13. ROOT / "fft" / "__init__.pyi",
  14. ROOT / "lib" / "__init__.pyi",
  15. ROOT / "linalg" / "__init__.pyi",
  16. ROOT / "ma" / "__init__.pyi",
  17. ROOT / "matrixlib" / "__init__.pyi",
  18. ROOT / "polynomial" / "__init__.pyi",
  19. ROOT / "random" / "__init__.pyi",
  20. ROOT / "testing" / "__init__.pyi",
  21. ]
  22. if sys.version_info < (3, 12):
  23. FILES += [ROOT / "distutils" / "__init__.pyi"]
  24. class TestIsFile:
  25. def test_isfile(self):
  26. """Test if all ``.pyi`` files are properly installed."""
  27. for file in FILES:
  28. assert_(os.path.isfile(file))