cpython: f2991b52157e (original) (raw)
--- a/Lib/test/test_calendar.py +++ b/Lib/test/test_calendar.py @@ -471,6 +471,9 @@ class LeapdaysTestCase(unittest.TestCase self.assertEqual(calendar.leapdays(1997,2020), 5) +def conv(s):
+ class CommandLineTestCase(unittest.TestCase): def run_ok(self, *args): return assert_python_ok('-m', 'calendar', *args)[1] @@ -501,17 +504,17 @@ class CommandLineTestCase(unittest.TestC def test_output_year(self): stdout = self.run_ok('2004')
self.assertEqual(stdout.strip(), result_2004_text.strip())[](#l1.17)
self.assertEqual(stdout.strip(), conv(result_2004_text).strip())[](#l1.18)
def test_output_month(self): stdout = self.run_ok('2004', '1')
self.assertEqual(stdout.strip(), result_2004_01_text.strip())[](#l1.22)
self.assertEqual(stdout.strip(), conv(result_2004_01_text).strip())[](#l1.23)
def test_option_encoding(self): self.assertFailure('-e') self.assertFailure('--encoding') stdout = self.run_ok('--encoding', 'rot-13', '2004')
self.assertEqual(stdout.strip(), result_2004_text.encode('rot-13').strip())[](#l1.29)
self.assertEqual(stdout.strip(), conv(result_2004_text.encode('rot-13')).strip())[](#l1.30)
def test_option_locale(self): self.assertFailure('-L') @@ -543,7 +546,7 @@ class CommandLineTestCase(unittest.TestC self.assertFailure('--lines') self.assertFailure('-l', 'spam') stdout = self.run_ok('--lines', '2', '2004')
self.assertIn('December\n\nMo Tu We', stdout)[](#l1.38)
self.assertIn(conv('December\n\nMo Tu We'), stdout)[](#l1.39)
def test_option_spacing(self): self.assertFailure('-s') @@ -557,14 +560,14 @@ class CommandLineTestCase(unittest.TestC self.assertFailure('--month') self.assertFailure('-m', 'spam') stdout = self.run_ok('--months', '1', '2004')
self.assertIn('\nMo Tu We Th Fr Sa Su\n', stdout)[](#l1.47)
self.assertIn(conv('\nMo Tu We Th Fr Sa Su\n'), stdout)[](#l1.48)
def test_option_type(self): self.assertFailure('-t') self.assertFailure('--type') self.assertFailure('-t', 'spam') stdout = self.run_ok('--type', 'text', '2004')
self.assertEqual(stdout.strip(), result_2004_text.strip())[](#l1.55)
self.assertEqual(stdout.strip(), conv(result_2004_text).strip())[](#l1.56) stdout = self.run_ok('--type', 'html', '2004')[](#l1.57) self.assertEqual(stdout[:6], b'<?xml ')[](#l1.58) self.assertIn(b'<title>Calendar for 2004</title>', stdout)[](#l1.59)
@@ -579,7 +582,7 @@ class CommandLineTestCase(unittest.TestC def test_html_output_year_encoding(self): stdout = self.run_ok('-t', 'html', '--encoding', 'ascii', '2004')
self.assertEqual(stdout.strip(), result_2004_html.strip())[](#l1.64)
self.assertEqual(stdout.strip(), conv(result_2004_html).strip())[](#l1.65)
def test_html_output_year_css(self): self.assertFailure('-t', 'html', '-c')