ENH: add support for pathlib.Path inputs in import_file

astropy · Sep 5, 2024 · e32444a · e32444a
1 parent e6996c5
commit e32444a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/extension_helpers/_utils.py b/extension_helpers/_utils.py
@@ -130,7 +130,7 @@ def import_file(filename, name=None):
     if not os.path.exists(filename):
         raise ImportError(f"Could not import file {filename}")
 
-    loader = import_machinery.SourceFileLoader(name, filename)
+    loader = import_machinery.SourceFileLoader(name, os.fspath(filename))
     spec = spec_from_file_location(name, filename)
     mod = module_from_spec(spec)
     loader.exec_module(mod)

diff --git a/extension_helpers/tests/test_utils.py b/extension_helpers/tests/test_utils.py
@@ -5,15 +5,15 @@
 
 
 def test_import_file(tmp_path):
-    filename = str(tmp_path / "spam.py")
+    filename = tmp_path / "spam.py"
     with open(filename, "w") as f:
         f.write("magic = 12345")
     module = import_file(filename)
     assert module.magic == 12345
 
 
 def test_write_if_different(tmp_path):
-    filename = str(tmp_path / "test.txt")
+    filename = tmp_path / "test.txt"
     write_if_different(filename, b"abc")
     time1 = os.path.getmtime(filename)
     time.sleep(0.01)