Add test for PYTHON_GIL in test_cmd_line

swtaarrs · swtaarrs · commit 4f43a87ea9b8 · 2024-03-05T14:02:24.000-08:00
diff --git a/Lib/test/test_cmd_line.py b/Lib/test/test_cmd_line.py
@@ -871,6 +871,38 @@ def test_pythondevmode_env(self):
         self.assertEqual(proc.stdout.rstrip(), 'True')
         self.assertEqual(proc.returncode, 0, proc)
 
+    @unittest.skipUnless(support.Py_GIL_DISABLED,
+                         "PYTHON_GIL only supported in Py_GIL_DISABLED builds")
+    def test_python_gil_env(self):
+        code = """if 1:
+            import _testinternalcapi
+            print(_testinternalcapi.get_configs()['config'].get('enable_gil'))
+            """
+        args = [sys.executable, '-c', code]
+        env = dict(os.environ)
+        env.pop('PYTHON_GIL', None)
+
+        def run():
+            return subprocess.run(args, stdout=subprocess.PIPE,
+                                  stderr=subprocess.PIPE, text=True, env=env)
+
+        proc = run()
+        self.assertEqual(proc.returncode, 0, proc)
+        self.assertEqual(proc.stdout.rstrip(), '0')
+        self.assertEqual(proc.stderr, '')
+
+        env['PYTHON_GIL'] = '0'
+        proc = run()
+        self.assertEqual(proc.returncode, 0, proc)
+        self.assertEqual(proc.stdout.rstrip(), '1')
+        self.assertEqual(proc.stderr, '')
+
+        env['PYTHON_GIL'] = '1'
+        proc = run()
+        self.assertEqual(proc.returncode, 0, proc)
+        self.assertEqual(proc.stdout.rstrip(), '2')
+        self.assertEqual(proc.stderr, '')
+
     @unittest.skipUnless(sys.platform == 'win32',
                          'bpo-32457 only applies on Windows')
     def test_argv0_normalization(self):
