test_isfile.py 812 B

123456789101112131415161718192021222324252627282930
  1. import os
  2. from pathlib import Path
  3. import numpy as np
  4. from numpy.testing import assert_
  5. ROOT = Path(np.__file__).parents[0]
  6. FILES = [
  7. ROOT / "py.typed",
  8. ROOT / "__init__.pyi",
  9. ROOT / "ctypeslib.pyi",
  10. ROOT / "core" / "__init__.pyi",
  11. ROOT / "distutils" / "__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. class TestIsFile:
  23. def test_isfile(self):
  24. """Test if all ``.pyi`` files are properly installed."""
  25. for file in FILES:
  26. assert_(os.path.isfile(file))