Fix tests

youknowone · youknowone · commit e170a998318b · 2023-08-30T18:28:34.000+09:00
diff --git a/Lib/test/test_cmd_line.py b/Lib/test/test_cmd_line.py
@@ -1008,26 +1008,6 @@ def test_sys_flags_not_set(self):
             PYTHONSAFEPATH="1",
         )
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
-    def test_pythonsafepath_env(self):
-        # Test the PYTHONSAFEPATH environment variable
-        code = "import sys; print(sys.flags.safe_path)"
-        env = dict(os.environ)
-        env.pop('PYTHONSAFEPATH', None)
-        args = (sys.executable, '-P', code)
-
-        proc = subprocess.run(args, stdout=subprocess.PIPE,
-                              universal_newlines=True, env=env)
-        self.assertEqual(proc.stdout.rstrip(), 'False')
-        self.assertEqual(proc.returncode, 0, proc)
-
-        env['PYTHONSAFEPATH'] = '1'
-        proc = subprocess.run(args, stdout=subprocess.PIPE,
-                              universal_newlines=True, env=env)
-        self.assertEqual(proc.stdout.rstrip(), 'True')
-        self.assertEqual(proc.returncode, 0, proc)
-
 class SyntaxErrorTests(unittest.TestCase):
     def check_string(self, code):
         proc = subprocess.run([sys.executable, "-"], input=code,
diff --git a/Lib/test/test_support.py b/Lib/test/test_support.py
@@ -513,8 +513,6 @@ def check_options(self, args, func, expected=None):
         self.assertEqual(proc.stdout.rstrip(), repr(expected))
         self.assertEqual(proc.returncode, 0)
 
-    # TODO: RUSTPYTHON
-    # @unittest.expectedFailure
     def test_args_from_interpreter_flags(self):
         # Test test.support.args_from_interpreter_flags()
         for opts in (
diff --git a/extra_tests/snippets/stdlib_sys.py b/extra_tests/snippets/stdlib_sys.py
@@ -1,4 +1,6 @@
 import sys
+import os
+import subprocess
 
 from testutils import assert_raises
 
@@ -105,4 +107,23 @@ def recursive_call(n):
     sys.set_int_max_str_digits(1)
 
 sys.set_int_max_str_digits(1000)
-assert sys.get_int_max_str_digits() == 1000
+assert sys.get_int_max_str_digits() == 1000
+
+# Test the PYTHONSAFEPATH environment variable
+code = "import sys; print(sys.flags.safe_path)"
+env = dict(os.environ)
+env.pop('PYTHONSAFEPATH', None)
+args = (sys.executable, '-P', '-c', code)
+
+proc = subprocess.run(
+    args, stdout=subprocess.PIPE,
+    universal_newlines=True, env=env)
+assert proc.stdout.rstrip() == 'True', proc
+assert proc.returncode == 0, proc
+
+env['PYTHONSAFEPATH'] = '1'
+proc = subprocess.run(
+    args, stdout=subprocess.PIPE,
+    universal_newlines=True, env=env)
+assert proc.stdout.rstrip() == 'True'
+assert proc.returncode == 0, proc
