Merge pull request #65 from mkoeppe/ccompiler__has_function__output_dir

jaraco · web-flow · commit f1b0a2bff47e · 2021-11-09T17:42:35.000-05:00
CCompiler.has_function: Do not fail if self.outputdir is set
diff --git a/distutils/ccompiler.py b/distutils/ccompiler.py
@@ -802,7 +802,7 @@ def has_function(self, funcname, includes=None, include_dirs=None,
         except (LinkError, TypeError):
             return False
         else:
-            os.remove("a.out")
+            os.remove(os.path.join(self.output_dir or '', "a.out"))
         finally:
             for fn in objects:
                 os.remove(fn)
diff --git a/distutils/tests/test_unixccompiler.py b/distutils/tests/test_unixccompiler.py
@@ -232,6 +232,13 @@ def gcvs(*args, _orig=sysconfig.get_config_vars):
             sysconfig.customize_compiler(self.cc)
         self.assertEqual(self.cc.linker_so[0], 'my_ld')
 
+    def test_has_function(self):
+        # Issue https://github.com/pypa/distutils/issues/64:
+        # ensure that setting output_dir does not raise
+        # FileNotFoundError: [Errno 2] No such file or directory: 'a.out'
+        self.cc.output_dir = 'scratch'
+        self.cc.has_function('abort', includes=['stdlib.h'])
+
 
 def test_suite():
     return unittest.makeSuite(UnixCCompilerTestCase)
