Lib/test/test_cmd_line.py

   1 # Tests invocation of the interpreter with various command line arguments
   2 # All tests are executed with environment variables ignored
   3 # See test_cmd_line_script.py for testing of script execution
   4
   5 import test.test_support, unittest
   6 import sys
   7 from test.script_helper import spawn_python, kill_python, python_exit_code
   8
   9
  10 class CmdLineTest(unittest.TestCase):
  11     def start_python(self, *args):
  12         p = spawn_python(*args)
  13         return kill_python(p)
  14
  15     def exit_code(self, *args):
  16         return python_exit_code(*args)
  17
  18     def test_directories(self):
  19         self.assertNotEqual(self.exit_code('.'), 0)
  20         self.assertNotEqual(self.exit_code('< .'), 0)
  21
  22     def verify_valid_flag(self, cmd_line):
  23         data = self.start_python(cmd_line)
  24         self.assertTrue(data == '' or data.endswith('\n'))
  25         self.assertNotIn('Traceback', data)
  26
  27     def test_optimize(self):
  28         self.verify_valid_flag('-O')
  29         self.verify_valid_flag('-OO')
  30
  31     def test_q(self):
  32         self.verify_valid_flag('-Qold')
  33         self.verify_valid_flag('-Qnew')
  34         self.verify_valid_flag('-Qwarn')
  35         self.verify_valid_flag('-Qwarnall')
  36
  37     def test_site_flag(self):
  38         self.verify_valid_flag('-S')
  39
  40     def test_usage(self):
  41         self.assertIn('usage', self.start_python('-h'))
  42
  43     def test_version(self):
  44         version = 'Python %d.%d' % sys.version_info[:2]
  45         self.assertTrue(self.start_python('-V').startswith(version))
  46
  47     def test_run_module(self):
  48         # Test expected operation of the '-m' switch
  49         # Switch needs an argument
  50         self.assertNotEqual(self.exit_code('-m'), 0)
  51         # Check we get an error for a nonexistent module
  52         self.assertNotEqual(
  53             self.exit_code('-m', 'fnord43520xyz'),
  54             0)
  55         # Check the runpy module also gives an error for
  56         # a nonexistent module
  57         self.assertNotEqual(
  58             self.exit_code('-m', 'runpy', 'fnord43520xyz'),
  59             0)
  60         # All good if module is located and run successfully
  61         self.assertEqual(
  62             self.exit_code('-m', 'timeit', '-n', '1'),
  63             0)
  64
  65     def test_run_module_bug1764407(self):
  66         # -m and -i need to play well together
  67         # Runs the timeit module and checks the __main__
  68         # namespace has been populated appropriately
  69         p = spawn_python('-i', '-m', 'timeit', '-n', '1')
  70         p.stdin.write('Timer\n')
  71         p.stdin.write('exit()\n')
  72         data = kill_python(p)
  73         self.assertTrue(data.startswith('1 loop'))
  74         self.assertIn('__main__.Timer', data)
  75
  76     def test_run_code(self):
  77         # Test expected operation of the '-c' switch
  78         # Switch needs an argument
  79         self.assertNotEqual(self.exit_code('-c'), 0)
  80         # Check we get an error for an uncaught exception
  81         self.assertNotEqual(
  82             self.exit_code('-c', 'raise Exception'),
  83             0)
  84         # All good if execution is successful
  85         self.assertEqual(
  86             self.exit_code('-c', 'pass'),
  87             0)
  88
  89
  90 def test_main():
  91     test.test_support.run_unittest(CmdLineTest)
  92     test.test_support.reap_children()
  93
  94 if __name__ == "__main__":
  95     test_main()