diff -r 5c7697f28ee4 Lib/test/test_calendar.py
--- a/Lib/test/test_calendar.py Mon Sep 02 08:57:21 2013 -0700
+++ b/Lib/test/test_calendar.py Thu Sep 05 08:45:39 2013 -0400
@@ -3,6 +3,8 @@
from test import support
from test.script_helper import assert_python_ok
+from io import StringIO
+import unittest.mock
import time
import locale
import sys
@@ -399,6 +401,27 @@
'
January |
'
)
+ def test_prweek(self):
+ with support.captured_stdout() as out:
+ week = [(1,0), (2,1), (3,2), (4,3), (5,4), (6,5), (7,6)]
+ calendar.TextCalendar().prweek(week, 1)
+ self.assertEqual(out.getvalue().strip(), "1 2 3 4 5 6 7")
+
+ def test_prmonth(self):
+ with support.captured_stdout() as out:
+ calendar.TextCalendar().prmonth(2004, 1)
+ output = out.getvalue().strip()
+ self.assertEqual(output, result_2004_01_text.strip())
+
+ def test_pryear(self):
+ with support.captured_stdout() as out:
+ calendar.TextCalendar().pryear(2004)
+ self.assertEqual(out.getvalue().strip(), result_2004_text.strip())
+
+ def test_format(self):
+ with support.captured_stdout() as out:
+ calendar.format(["1", "2", "3"], colwidth=3, spacing=1)
+ self.assertEqual(out.getvalue().strip(), "1 2 3")
class CalendarTestCase(unittest.TestCase):
def test_isleap(self):