Add test

corona10 · Oct 1, 2023 · 2bec7f4 · 2bec7f4
1 parent 134ed9e
commit 2bec7f4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/Lib/os.py b/Lib/os.py
@@ -1137,6 +1137,7 @@ def add_dll_directory(path):
             nt._remove_dll_directory
         )
 
+
 if _exists('sched_getaffinity') and os._get_cpu_config() < 0:
     def process_cpu_count():
         """

diff --git a/Lib/test/test_cmd_line.py b/Lib/test/test_cmd_line.py
@@ -902,11 +902,15 @@ def test_cpu_count(self):
         code = "import os; print(os.cpu_count(), os.process_cpu_count())"
         res = assert_python_ok('-X', 'cpu_count=4321', '-c', code)
         self.assertEqual(self.res2int(res), (4321, 4321))
+        res = assert_python_ok('-c', code, PYTHONCPUCOUNT='1234')
+        self.assertEqual(self.res2int(res), (1234, 1234))
 
     def test_cpu_count_default(self):
         code = "import os; print(os.cpu_count(), os.process_cpu_count())"
         res = assert_python_ok('-X', 'cpu_count=default', '-c', code)
         self.assertEqual(self.res2int(res), (os.cpu_count(), os.process_cpu_count()))
+        res = assert_python_ok('-X', 'cpu_count=default', '-c', code, PYTHONCPUCOUNT='1234')
+        self.assertEqual(self.res2int(res), (os.cpu_count(), os.process_cpu_count()))
 
     def res2int(self, res):
         out = res.out.strip().decode("utf-8")
-Original file line number
+Diff line change
@@ Expand Up / @@ -1137,6 +1137,7 @@ def add_dll_directory(path): @@
                 nt._remove_dll_directory
             )
     if _exists('sched_getaffinity') and os._get_cpu_config() < 0:
         def process_cpu_count():
             """
@@ Expand Down @@