pythongh-69714: Add more test cases to calendar cmdline

bxsx · Jun 9, 2022 · 238c527 · 238c527
1 parent 636b4e1
commit 238c527
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/Lib/test/test_calendar.py b/Lib/test/test_calendar.py
@@ -934,6 +934,15 @@ def test_illegal_arguments(self):
         self.assertFailure('2004', 'spam')
         self.assertFailure('-t', 'html', '2004', '1')
 
+    def test_too_many_arguments(self):
+        stderr = self.assertCmdFails('--type', 'html', '2004', '5')[2]
+        self.assertIn(b'error', stderr)
+        self.assertIn(b'incorrect number of arguments', stderr)
+
+        stderr = self.assertCLIFails('--type', 'html', '2004', '5')
+        self.assertIn(b'error', stderr)
+        self.assertIn(b'incorrect number of arguments', stderr)
+
     def test_output_current_year(self):
         for run in self.runners:
             output = run()