Skip to content

C
cpython
Dal (branch)/etiket değiştir
Dosya bul
Normal viewHistoryPermalink
test_calendar.py 44 KB
Newer Older
Fred Drake's avatar
Very small test suite for the calendar module, mostly to check a constraint
d077ca1e
 
Fred Drake committed
1 2 3 
import calendar
import unittest
Benjamin Peterson's avatar
#2621 rename test.test_support to test.support
ee8712cd
 
Benjamin Peterson committed
4 
from test import support
Berker Peksag's avatar
Issue #9517: Move script_helper to the support package.
ce643913
 
Berker Peksag committed
5 
from test.support.script_helper import assert_python_ok, assert_python_failure
Alexander Belopolsky's avatar
Issue 6280: Tests and simpler implementation for calendar.timegm
43ca710a
 
Alexander Belopolsky committed
6 
import time
Georg Brandl's avatar
#10092: Properly reset locale in Locale*Calendar classes. The context manager…
7004bd1a
 
Georg Brandl committed
7 
import locale
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
8 
import sys
Ezio Melotti's avatar
#15421: fix an OverflowError in Calendar.itermonthdates() after…
85710a40
 
Ezio Melotti committed
9 
import datetime
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
10 
import os
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
11
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
12 
result_2004_01_text = """\
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
13 14 15 16 17 18 19 20 
    January 2004
Mo Tu We Th Fr Sa Su
          1  2  3  4
 5  6  7  8  9 10 11
12 13 14 15 16 17 18
19 20 21 22 23 24 25
26 27 28 29 30 31
"""
Fred Drake's avatar
Very small test suite for the calendar module, mostly to check a constraint
d077ca1e
 
Fred Drake committed
21
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
22 
result_2004_text = """\
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 
                                  2004

      January                   February                   March
Mo Tu We Th Fr Sa Su      Mo Tu We Th Fr Sa Su      Mo Tu We Th Fr Sa Su
          1  2  3  4                         1       1  2  3  4  5  6  7
 5  6  7  8  9 10 11       2  3  4  5  6  7  8       8  9 10 11 12 13 14
12 13 14 15 16 17 18       9 10 11 12 13 14 15      15 16 17 18 19 20 21
19 20 21 22 23 24 25      16 17 18 19 20 21 22      22 23 24 25 26 27 28
26 27 28 29 30 31         23 24 25 26 27 28 29      29 30 31

       April                      May                       June
Mo Tu We Th Fr Sa Su      Mo Tu We Th Fr Sa Su      Mo Tu We Th Fr Sa Su
          1  2  3  4                      1  2          1  2  3  4  5  6
 5  6  7  8  9 10 11       3  4  5  6  7  8  9       7  8  9 10 11 12 13
12 13 14 15 16 17 18      10 11 12 13 14 15 16      14 15 16 17 18 19 20
19 20 21 22 23 24 25      17 18 19 20 21 22 23      21 22 23 24 25 26 27
26 27 28 29 30            24 25 26 27 28 29 30      28 29 30
                          31

        July                     August                  September
Mo Tu We Th Fr Sa Su      Mo Tu We Th Fr Sa Su      Mo Tu We Th Fr Sa Su
          1  2  3  4                         1             1  2  3  4  5
 5  6  7  8  9 10 11       2  3  4  5  6  7  8       6  7  8  9 10 11 12
12 13 14 15 16 17 18       9 10 11 12 13 14 15      13 14 15 16 17 18 19
19 20 21 22 23 24 25      16 17 18 19 20 21 22      20 21 22 23 24 25 26
26 27 28 29 30 31         23 24 25 26 27 28 29      27 28 29 30
                          30 31

      October                   November                  December
Mo Tu We Th Fr Sa Su      Mo Tu We Th Fr Sa Su      Mo Tu We Th Fr Sa Su
             1  2  3       1  2  3  4  5  6  7             1  2  3  4  5
 4  5  6  7  8  9 10       8  9 10 11 12 13 14       6  7  8  9 10 11 12
11 12 13 14 15 16 17      15 16 17 18 19 20 21      13 14 15 16 17 18 19
18 19 20 21 22 23 24      22 23 24 25 26 27 28      20 21 22 23 24 25 26
25 26 27 28 29 30 31      29 30                     27 28 29 30 31
"""
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
60 
result_2004_html = """\
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
61 
<?xml version="1.0" encoding="%(e)s"?>
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
62 63 64 
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
<html>
<head>
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
65 
<meta http-equiv="Content-Type" content="text/html; charset=%(e)s" />
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
66 
<link rel="stylesheet" type="text/css" href="calendar.css" />
Thomas Wouters's avatar
Merged revisions 57620-57771 via svnmerge from
47b49bf6
 
Thomas Wouters committed
67 
<title>Calendar for 2004</title>
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 
</head>
<body>
<table border="0" cellpadding="0" cellspacing="0" class="year">
<tr><th colspan="3" class="year">2004</th></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">January</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="thu">1</td><td class="fri">2</td><td class="sat">3</td><td class="sun">4</td></tr>
<tr><td class="mon">5</td><td class="tue">6</td><td class="wed">7</td><td class="thu">8</td><td class="fri">9</td><td class="sat">10</td><td class="sun">11</td></tr>
<tr><td class="mon">12</td><td class="tue">13</td><td class="wed">14</td><td class="thu">15</td><td class="fri">16</td><td class="sat">17</td><td class="sun">18</td></tr>
<tr><td class="mon">19</td><td class="tue">20</td><td class="wed">21</td><td class="thu">22</td><td class="fri">23</td><td class="sat">24</td><td class="sun">25</td></tr>
<tr><td class="mon">26</td><td class="tue">27</td><td class="wed">28</td><td class="thu">29</td><td class="fri">30</td><td class="sat">31</td><td class="noday">&nbsp;</td></tr>
</table>
</td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">February</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="sun">1</td></tr>
<tr><td class="mon">2</td><td class="tue">3</td><td class="wed">4</td><td class="thu">5</td><td class="fri">6</td><td class="sat">7</td><td class="sun">8</td></tr>
<tr><td class="mon">9</td><td class="tue">10</td><td class="wed">11</td><td class="thu">12</td><td class="fri">13</td><td class="sat">14</td><td class="sun">15</td></tr>
<tr><td class="mon">16</td><td class="tue">17</td><td class="wed">18</td><td class="thu">19</td><td class="fri">20</td><td class="sat">21</td><td class="sun">22</td></tr>
<tr><td class="mon">23</td><td class="tue">24</td><td class="wed">25</td><td class="thu">26</td><td class="fri">27</td><td class="sat">28</td><td class="sun">29</td></tr>
</table>
</td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">March</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="mon">1</td><td class="tue">2</td><td class="wed">3</td><td class="thu">4</td><td class="fri">5</td><td class="sat">6</td><td class="sun">7</td></tr>
<tr><td class="mon">8</td><td class="tue">9</td><td class="wed">10</td><td class="thu">11</td><td class="fri">12</td><td class="sat">13</td><td class="sun">14</td></tr>
<tr><td class="mon">15</td><td class="tue">16</td><td class="wed">17</td><td class="thu">18</td><td class="fri">19</td><td class="sat">20</td><td class="sun">21</td></tr>
<tr><td class="mon">22</td><td class="tue">23</td><td class="wed">24</td><td class="thu">25</td><td class="fri">26</td><td class="sat">27</td><td class="sun">28</td></tr>
<tr><td class="mon">29</td><td class="tue">30</td><td class="wed">31</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td></tr>
</table>
</td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">April</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="thu">1</td><td class="fri">2</td><td class="sat">3</td><td class="sun">4</td></tr>
<tr><td class="mon">5</td><td class="tue">6</td><td class="wed">7</td><td class="thu">8</td><td class="fri">9</td><td class="sat">10</td><td class="sun">11</td></tr>
<tr><td class="mon">12</td><td class="tue">13</td><td class="wed">14</td><td class="thu">15</td><td class="fri">16</td><td class="sat">17</td><td class="sun">18</td></tr>
<tr><td class="mon">19</td><td class="tue">20</td><td class="wed">21</td><td class="thu">22</td><td class="fri">23</td><td class="sat">24</td><td class="sun">25</td></tr>
<tr><td class="mon">26</td><td class="tue">27</td><td class="wed">28</td><td class="thu">29</td><td class="fri">30</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td></tr>
</table>
</td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">May</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="sat">1</td><td class="sun">2</td></tr>
<tr><td class="mon">3</td><td class="tue">4</td><td class="wed">5</td><td class="thu">6</td><td class="fri">7</td><td class="sat">8</td><td class="sun">9</td></tr>
<tr><td class="mon">10</td><td class="tue">11</td><td class="wed">12</td><td class="thu">13</td><td class="fri">14</td><td class="sat">15</td><td class="sun">16</td></tr>
<tr><td class="mon">17</td><td class="tue">18</td><td class="wed">19</td><td class="thu">20</td><td class="fri">21</td><td class="sat">22</td><td class="sun">23</td></tr>
<tr><td class="mon">24</td><td class="tue">25</td><td class="wed">26</td><td class="thu">27</td><td class="fri">28</td><td class="sat">29</td><td class="sun">30</td></tr>
<tr><td class="mon">31</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td></tr>
</table>
</td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">June</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="tue">1</td><td class="wed">2</td><td class="thu">3</td><td class="fri">4</td><td class="sat">5</td><td class="sun">6</td></tr>
<tr><td class="mon">7</td><td class="tue">8</td><td class="wed">9</td><td class="thu">10</td><td class="fri">11</td><td class="sat">12</td><td class="sun">13</td></tr>
<tr><td class="mon">14</td><td class="tue">15</td><td class="wed">16</td><td class="thu">17</td><td class="fri">18</td><td class="sat">19</td><td class="sun">20</td></tr>
<tr><td class="mon">21</td><td class="tue">22</td><td class="wed">23</td><td class="thu">24</td><td class="fri">25</td><td class="sat">26</td><td class="sun">27</td></tr>
<tr><td class="mon">28</td><td class="tue">29</td><td class="wed">30</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td></tr>
</table>
</td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">July</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="thu">1</td><td class="fri">2</td><td class="sat">3</td><td class="sun">4</td></tr>
<tr><td class="mon">5</td><td class="tue">6</td><td class="wed">7</td><td class="thu">8</td><td class="fri">9</td><td class="sat">10</td><td class="sun">11</td></tr>
<tr><td class="mon">12</td><td class="tue">13</td><td class="wed">14</td><td class="thu">15</td><td class="fri">16</td><td class="sat">17</td><td class="sun">18</td></tr>
<tr><td class="mon">19</td><td class="tue">20</td><td class="wed">21</td><td class="thu">22</td><td class="fri">23</td><td class="sat">24</td><td class="sun">25</td></tr>
<tr><td class="mon">26</td><td class="tue">27</td><td class="wed">28</td><td class="thu">29</td><td class="fri">30</td><td class="sat">31</td><td class="noday">&nbsp;</td></tr>
</table>
</td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">August</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="sun">1</td></tr>
<tr><td class="mon">2</td><td class="tue">3</td><td class="wed">4</td><td class="thu">5</td><td class="fri">6</td><td class="sat">7</td><td class="sun">8</td></tr>
<tr><td class="mon">9</td><td class="tue">10</td><td class="wed">11</td><td class="thu">12</td><td class="fri">13</td><td class="sat">14</td><td class="sun">15</td></tr>
<tr><td class="mon">16</td><td class="tue">17</td><td class="wed">18</td><td class="thu">19</td><td class="fri">20</td><td class="sat">21</td><td class="sun">22</td></tr>
<tr><td class="mon">23</td><td class="tue">24</td><td class="wed">25</td><td class="thu">26</td><td class="fri">27</td><td class="sat">28</td><td class="sun">29</td></tr>
<tr><td class="mon">30</td><td class="tue">31</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td></tr>
</table>
</td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">September</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="wed">1</td><td class="thu">2</td><td class="fri">3</td><td class="sat">4</td><td class="sun">5</td></tr>
<tr><td class="mon">6</td><td class="tue">7</td><td class="wed">8</td><td class="thu">9</td><td class="fri">10</td><td class="sat">11</td><td class="sun">12</td></tr>
<tr><td class="mon">13</td><td class="tue">14</td><td class="wed">15</td><td class="thu">16</td><td class="fri">17</td><td class="sat">18</td><td class="sun">19</td></tr>
<tr><td class="mon">20</td><td class="tue">21</td><td class="wed">22</td><td class="thu">23</td><td class="fri">24</td><td class="sat">25</td><td class="sun">26</td></tr>
<tr><td class="mon">27</td><td class="tue">28</td><td class="wed">29</td><td class="thu">30</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td></tr>
</table>
</td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">October</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="fri">1</td><td class="sat">2</td><td class="sun">3</td></tr>
<tr><td class="mon">4</td><td class="tue">5</td><td class="wed">6</td><td class="thu">7</td><td class="fri">8</td><td class="sat">9</td><td class="sun">10</td></tr>
<tr><td class="mon">11</td><td class="tue">12</td><td class="wed">13</td><td class="thu">14</td><td class="fri">15</td><td class="sat">16</td><td class="sun">17</td></tr>
<tr><td class="mon">18</td><td class="tue">19</td><td class="wed">20</td><td class="thu">21</td><td class="fri">22</td><td class="sat">23</td><td class="sun">24</td></tr>
<tr><td class="mon">25</td><td class="tue">26</td><td class="wed">27</td><td class="thu">28</td><td class="fri">29</td><td class="sat">30</td><td class="sun">31</td></tr>
</table>
</td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">November</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="mon">1</td><td class="tue">2</td><td class="wed">3</td><td class="thu">4</td><td class="fri">5</td><td class="sat">6</td><td class="sun">7</td></tr>
<tr><td class="mon">8</td><td class="tue">9</td><td class="wed">10</td><td class="thu">11</td><td class="fri">12</td><td class="sat">13</td><td class="sun">14</td></tr>
<tr><td class="mon">15</td><td class="tue">16</td><td class="wed">17</td><td class="thu">18</td><td class="fri">19</td><td class="sat">20</td><td class="sun">21</td></tr>
<tr><td class="mon">22</td><td class="tue">23</td><td class="wed">24</td><td class="thu">25</td><td class="fri">26</td><td class="sat">27</td><td class="sun">28</td></tr>
<tr><td class="mon">29</td><td class="tue">30</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td></tr>
</table>
</td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
<tr><th colspan="7" class="month">December</th></tr>
<tr><th class="mon">Mon</th><th class="tue">Tue</th><th class="wed">Wed</th><th class="thu">Thu</th><th class="fri">Fri</th><th class="sat">Sat</th><th class="sun">Sun</th></tr>
<tr><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td><td class="wed">1</td><td class="thu">2</td><td class="fri">3</td><td class="sat">4</td><td class="sun">5</td></tr>
<tr><td class="mon">6</td><td class="tue">7</td><td class="wed">8</td><td class="thu">9</td><td class="fri">10</td><td class="sat">11</td><td class="sun">12</td></tr>
<tr><td class="mon">13</td><td class="tue">14</td><td class="wed">15</td><td class="thu">16</td><td class="fri">17</td><td class="sat">18</td><td class="sun">19</td></tr>
<tr><td class="mon">20</td><td class="tue">21</td><td class="wed">22</td><td class="thu">23</td><td class="fri">24</td><td class="sat">25</td><td class="sun">26</td></tr>
<tr><td class="mon">27</td><td class="tue">28</td><td class="wed">29</td><td class="thu">30</td><td class="fri">31</td><td class="noday">&nbsp;</td><td class="noday">&nbsp;</td></tr>
</table>
</td></tr></table></body>
</html>
"""
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 
result_2004_days = [
    [[[0, 0, 0, 1, 2, 3, 4],
      [5, 6, 7, 8, 9, 10, 11],
      [12, 13, 14, 15, 16, 17, 18],
      [19, 20, 21, 22, 23, 24, 25],
      [26, 27, 28, 29, 30, 31, 0]],
     [[0, 0, 0, 0, 0, 0, 1],
      [2, 3, 4, 5, 6, 7, 8],
      [9, 10, 11, 12, 13, 14, 15],
      [16, 17, 18, 19, 20, 21, 22],
      [23, 24, 25, 26, 27, 28, 29]],
     [[1, 2, 3, 4, 5, 6, 7],
      [8, 9, 10, 11, 12, 13, 14],
      [15, 16, 17, 18, 19, 20, 21],
      [22, 23, 24, 25, 26, 27, 28],
      [29, 30, 31, 0, 0, 0, 0]]],
    [[[0, 0, 0, 1, 2, 3, 4],
      [5, 6, 7, 8, 9, 10, 11],
      [12, 13, 14, 15, 16, 17, 18],
      [19, 20, 21, 22, 23, 24, 25],
      [26, 27, 28, 29, 30, 0, 0]],
     [[0, 0, 0, 0, 0, 1, 2],
      [3, 4, 5, 6, 7, 8, 9],
      [10, 11, 12, 13, 14, 15, 16],
      [17, 18, 19, 20, 21, 22, 23],
      [24, 25, 26, 27, 28, 29, 30],
      [31, 0, 0, 0, 0, 0, 0]],
     [[0, 1, 2, 3, 4, 5, 6],
      [7, 8, 9, 10, 11, 12, 13],
      [14, 15, 16, 17, 18, 19, 20],
      [21, 22, 23, 24, 25, 26, 27],
      [28, 29, 30, 0, 0, 0, 0]]],
    [[[0, 0, 0, 1, 2, 3, 4],
      [5, 6, 7, 8, 9, 10, 11],
      [12, 13, 14, 15, 16, 17, 18],
      [19, 20, 21, 22, 23, 24, 25],
      [26, 27, 28, 29, 30, 31, 0]],
     [[0, 0, 0, 0, 0, 0, 1],
      [2, 3, 4, 5, 6, 7, 8],
      [9, 10, 11, 12, 13, 14, 15],
      [16, 17, 18, 19, 20, 21, 22],
      [23, 24, 25, 26, 27, 28, 29],
      [30, 31, 0, 0, 0, 0, 0]],
     [[0, 0, 1, 2, 3, 4, 5],
      [6, 7, 8, 9, 10, 11, 12],
      [13, 14, 15, 16, 17, 18, 19],
      [20, 21, 22, 23, 24, 25, 26],
      [27, 28, 29, 30, 0, 0, 0]]],
    [[[0, 0, 0, 0, 1, 2, 3],
      [4, 5, 6, 7, 8, 9, 10],
      [11, 12, 13, 14, 15, 16, 17],
      [18, 19, 20, 21, 22, 23, 24],
      [25, 26, 27, 28, 29, 30, 31]],
     [[1, 2, 3, 4, 5, 6, 7],
      [8, 9, 10, 11, 12, 13, 14],
      [15, 16, 17, 18, 19, 20, 21],
      [22, 23, 24, 25, 26, 27, 28],
      [29, 30, 0, 0, 0, 0, 0]],
     [[0, 0, 1, 2, 3, 4, 5],
      [6, 7, 8, 9, 10, 11, 12],
      [13, 14, 15, 16, 17, 18, 19],
      [20, 21, 22, 23, 24, 25, 26],
      [27, 28, 29, 30, 31, 0, 0]]]
]

result_2004_dates = \
    [[['12/29/03 12/30/03 12/31/03 01/01/04 01/02/04 01/03/04 01/04/04',
       '01/05/04 01/06/04 01/07/04 01/08/04 01/09/04 01/10/04 01/11/04',
       '01/12/04 01/13/04 01/14/04 01/15/04 01/16/04 01/17/04 01/18/04',
       '01/19/04 01/20/04 01/21/04 01/22/04 01/23/04 01/24/04 01/25/04',
       '01/26/04 01/27/04 01/28/04 01/29/04 01/30/04 01/31/04 02/01/04'],
      ['01/26/04 01/27/04 01/28/04 01/29/04 01/30/04 01/31/04 02/01/04',
       '02/02/04 02/03/04 02/04/04 02/05/04 02/06/04 02/07/04 02/08/04',
       '02/09/04 02/10/04 02/11/04 02/12/04 02/13/04 02/14/04 02/15/04',
       '02/16/04 02/17/04 02/18/04 02/19/04 02/20/04 02/21/04 02/22/04',
       '02/23/04 02/24/04 02/25/04 02/26/04 02/27/04 02/28/04 02/29/04'],
      ['03/01/04 03/02/04 03/03/04 03/04/04 03/05/04 03/06/04 03/07/04',
       '03/08/04 03/09/04 03/10/04 03/11/04 03/12/04 03/13/04 03/14/04',
       '03/15/04 03/16/04 03/17/04 03/18/04 03/19/04 03/20/04 03/21/04',
       '03/22/04 03/23/04 03/24/04 03/25/04 03/26/04 03/27/04 03/28/04',
       '03/29/04 03/30/04 03/31/04 04/01/04 04/02/04 04/03/04 04/04/04']],
     [['03/29/04 03/30/04 03/31/04 04/01/04 04/02/04 04/03/04 04/04/04',
       '04/05/04 04/06/04 04/07/04 04/08/04 04/09/04 04/10/04 04/11/04',
       '04/12/04 04/13/04 04/14/04 04/15/04 04/16/04 04/17/04 04/18/04',
       '04/19/04 04/20/04 04/21/04 04/22/04 04/23/04 04/24/04 04/25/04',
       '04/26/04 04/27/04 04/28/04 04/29/04 04/30/04 05/01/04 05/02/04'],
      ['04/26/04 04/27/04 04/28/04 04/29/04 04/30/04 05/01/04 05/02/04',
       '05/03/04 05/04/04 05/05/04 05/06/04 05/07/04 05/08/04 05/09/04',
       '05/10/04 05/11/04 05/12/04 05/13/04 05/14/04 05/15/04 05/16/04',
       '05/17/04 05/18/04 05/19/04 05/20/04 05/21/04 05/22/04 05/23/04',
       '05/24/04 05/25/04 05/26/04 05/27/04 05/28/04 05/29/04 05/30/04',
       '05/31/04 06/01/04 06/02/04 06/03/04 06/04/04 06/05/04 06/06/04'],
      ['05/31/04 06/01/04 06/02/04 06/03/04 06/04/04 06/05/04 06/06/04',
       '06/07/04 06/08/04 06/09/04 06/10/04 06/11/04 06/12/04 06/13/04',
       '06/14/04 06/15/04 06/16/04 06/17/04 06/18/04 06/19/04 06/20/04',
       '06/21/04 06/22/04 06/23/04 06/24/04 06/25/04 06/26/04 06/27/04',
       '06/28/04 06/29/04 06/30/04 07/01/04 07/02/04 07/03/04 07/04/04']],
     [['06/28/04 06/29/04 06/30/04 07/01/04 07/02/04 07/03/04 07/04/04',
       '07/05/04 07/06/04 07/07/04 07/08/04 07/09/04 07/10/04 07/11/04',
       '07/12/04 07/13/04 07/14/04 07/15/04 07/16/04 07/17/04 07/18/04',
       '07/19/04 07/20/04 07/21/04 07/22/04 07/23/04 07/24/04 07/25/04',
       '07/26/04 07/27/04 07/28/04 07/29/04 07/30/04 07/31/04 08/01/04'],
      ['07/26/04 07/27/04 07/28/04 07/29/04 07/30/04 07/31/04 08/01/04',
       '08/02/04 08/03/04 08/04/04 08/05/04 08/06/04 08/07/04 08/08/04',
       '08/09/04 08/10/04 08/11/04 08/12/04 08/13/04 08/14/04 08/15/04',
       '08/16/04 08/17/04 08/18/04 08/19/04 08/20/04 08/21/04 08/22/04',
       '08/23/04 08/24/04 08/25/04 08/26/04 08/27/04 08/28/04 08/29/04',
       '08/30/04 08/31/04 09/01/04 09/02/04 09/03/04 09/04/04 09/05/04'],
      ['08/30/04 08/31/04 09/01/04 09/02/04 09/03/04 09/04/04 09/05/04',
       '09/06/04 09/07/04 09/08/04 09/09/04 09/10/04 09/11/04 09/12/04',
       '09/13/04 09/14/04 09/15/04 09/16/04 09/17/04 09/18/04 09/19/04',
       '09/20/04 09/21/04 09/22/04 09/23/04 09/24/04 09/25/04 09/26/04',
       '09/27/04 09/28/04 09/29/04 09/30/04 10/01/04 10/02/04 10/03/04']],
     [['09/27/04 09/28/04 09/29/04 09/30/04 10/01/04 10/02/04 10/03/04',
       '10/04/04 10/05/04 10/06/04 10/07/04 10/08/04 10/09/04 10/10/04',
       '10/11/04 10/12/04 10/13/04 10/14/04 10/15/04 10/16/04 10/17/04',
       '10/18/04 10/19/04 10/20/04 10/21/04 10/22/04 10/23/04 10/24/04',
       '10/25/04 10/26/04 10/27/04 10/28/04 10/29/04 10/30/04 10/31/04'],
      ['11/01/04 11/02/04 11/03/04 11/04/04 11/05/04 11/06/04 11/07/04',
       '11/08/04 11/09/04 11/10/04 11/11/04 11/12/04 11/13/04 11/14/04',
       '11/15/04 11/16/04 11/17/04 11/18/04 11/19/04 11/20/04 11/21/04',
       '11/22/04 11/23/04 11/24/04 11/25/04 11/26/04 11/27/04 11/28/04',
       '11/29/04 11/30/04 12/01/04 12/02/04 12/03/04 12/04/04 12/05/04'],
      ['11/29/04 11/30/04 12/01/04 12/02/04 12/03/04 12/04/04 12/05/04',
       '12/06/04 12/07/04 12/08/04 12/09/04 12/10/04 12/11/04 12/12/04',
       '12/13/04 12/14/04 12/15/04 12/16/04 12/17/04 12/18/04 12/19/04',
       '12/20/04 12/21/04 12/22/04 12/23/04 12/24/04 12/25/04 12/26/04',
       '12/27/04 12/28/04 12/29/04 12/30/04 12/31/04 01/01/05 01/02/05']]]
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
314 315 316 

class OutputTestCase(unittest.TestCase):
    def normalize_calendar(self, s):
Mark Dickinson's avatar
Merged revisions 69846 via svnmerge from
934896dc
 
Mark Dickinson committed
317 
        # Filters out locale dependent strings
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
318 319 320 321 
        def neitherspacenordigit(c):
            return not c.isspace() and not c.isdigit()

        lines = []
Ezio Melotti's avatar
#13012: use splitlines(keepends=True/False) instead of splitlines(0/1).
d8b509b1
 
Ezio Melotti committed
322 
        for line in s.splitlines(keepends=False):
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
323 324 325 326 327 
            # Drop texts, as they are locale dependent
            if line and not filter(neitherspacenordigit, line):
                lines.append(line)
        return lines
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
328 329 330 
    def check_htmlcalendar_encoding(self, req, res):
        cal = calendar.HTMLCalendar()
        self.assertEqual(
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
331 332 
            cal.formatyearpage(2004, encoding=req),
            (result_2004_html % {'e': res}).encode(res)
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
333 334 
        )
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
335 336 337 338 339 340 341 342 
    def test_output(self):
        self.assertEqual(
            self.normalize_calendar(calendar.calendar(2004)),
            self.normalize_calendar(result_2004_text)
        )

    def test_output_textcalendar(self):
        self.assertEqual(
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
343 344 
            calendar.TextCalendar().formatyear(2004),
            result_2004_text
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
345 346 
        )
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
347 348 349 350 351 352 353 354 355 356 357 
    def test_output_htmlcalendar_encoding_ascii(self):
        self.check_htmlcalendar_encoding('ascii', 'ascii')

    def test_output_htmlcalendar_encoding_utf8(self):
        self.check_htmlcalendar_encoding('utf-8', 'utf-8')

    def test_output_htmlcalendar_encoding_default(self):
        self.check_htmlcalendar_encoding(None, sys.getdefaultencoding())

    def test_yeardatescalendar(self):
        def shrink(cal):
R David Murray's avatar
#14796: fix failure of new calendar test on windows.
bb756780
 
R David Murray committed
358 359 360 
            return [[[' '.join('{:02d}/{:02d}/{}'.format(
                                d.month, d.day, str(d.year)[-2:]) for d in z)
                            for z in y] for y in x] for x in cal]
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 
        self.assertEqual(
            shrink(calendar.Calendar().yeardatescalendar(2004)),
            result_2004_dates
        )

    def test_yeardayscalendar(self):
        self.assertEqual(
            calendar.Calendar().yeardayscalendar(2004),
            result_2004_days
        )

    def test_formatweekheader_short(self):
        self.assertEqual(
            calendar.TextCalendar().formatweekheader(2),
            'Mo Tu We Th Fr Sa Su'
        )

    def test_formatweekheader_long(self):
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
379 
        self.assertEqual(
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
380 381 382 383 384 385 386 
            calendar.TextCalendar().formatweekheader(9),
            '  Monday   Tuesday  Wednesday  Thursday '
            '  Friday   Saturday   Sunday '
        )

    def test_formatmonth(self):
        self.assertEqual(
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
387 388 
            calendar.TextCalendar().formatmonth(2004, 1),
            result_2004_01_text
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
389 390 391 392 393 394 395 396 397 398 399 400 
        )

    def test_formatmonthname_with_year(self):
        self.assertEqual(
            calendar.HTMLCalendar().formatmonthname(2004, 1, withyear=True),
            '<tr><th colspan="7" class="month">January 2004</th></tr>'
        )

    def test_formatmonthname_without_year(self):
        self.assertEqual(
            calendar.HTMLCalendar().formatmonthname(2004, 1, withyear=False),
            '<tr><th colspan="7" class="month">January</th></tr>'
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
401 402 
        )
Ezio Melotti's avatar
#18856: improve test coverage of the calendar module. Patch by Madison May.
87f09371
 
Ezio Melotti committed
403 404 405 406 
    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)
Serhiy Storchaka's avatar
Issue #28255: calendar.TextCalendar.prweek() no longer prints a space after
0595ed21
 
Serhiy Storchaka committed
407 
            self.assertEqual(out.getvalue(), " 1  2  3  4  5  6  7")
Ezio Melotti's avatar
#18856: improve test coverage of the calendar module. Patch by Madison May.
87f09371
 
Ezio Melotti committed
408 409 410 411 

    def test_prmonth(self):
        with support.captured_stdout() as out:
            calendar.TextCalendar().prmonth(2004, 1)
Serhiy Storchaka's avatar
Issue #28255: calendar.TextCalendar().prmonth() no longer prints a space
7ff51bd2
 
Serhiy Storchaka committed
412 
            self.assertEqual(out.getvalue(), result_2004_01_text)
Ezio Melotti's avatar
#18856: improve test coverage of the calendar module. Patch by Madison May.
87f09371
 
Ezio Melotti committed
413 414 415 416 

    def test_pryear(self):
        with support.captured_stdout() as out:
            calendar.TextCalendar().pryear(2004)
Serhiy Storchaka's avatar
Issue #28255: calendar.TextCalendar.prweek() no longer prints a space after
0595ed21
 
Serhiy Storchaka committed
417 
            self.assertEqual(out.getvalue(), result_2004_text)
Ezio Melotti's avatar
#18856: improve test coverage of the calendar module. Patch by Madison May.
87f09371
 
Ezio Melotti committed
418 419 420 421 422 

    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")
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
423
Fred Drake's avatar
Very small test suite for the calendar module, mostly to check a constraint
d077ca1e
 
Fred Drake committed
424 425 426 427 428 429 430 431 432 433 434 435 
class CalendarTestCase(unittest.TestCase):
    def test_isleap(self):
        # Make sure that the return is right for a few years, and
        # ensure that the return values are 1 or 0, not just true or
        # false (see SF bug #485794).  Specific additional tests may
        # be appropriate; this tests a single "cycle".
        self.assertEqual(calendar.isleap(2000), 1)
        self.assertEqual(calendar.isleap(2001), 0)
        self.assertEqual(calendar.isleap(2002), 0)
        self.assertEqual(calendar.isleap(2003), 0)

    def test_setfirstweekday(self):
Guido van Rossum's avatar
Restructure comparison dramatically. There is no longer a default
47b9ff6b
 
Guido van Rossum committed
436 
        self.assertRaises(TypeError, calendar.setfirstweekday, 'flabber')
Fred Drake's avatar
Very small test suite for the calendar module, mostly to check a constraint
d077ca1e
 
Fred Drake committed
437 438 439 440 441 442 443 444 445 
        self.assertRaises(ValueError, calendar.setfirstweekday, -1)
        self.assertRaises(ValueError, calendar.setfirstweekday, 200)
        orig = calendar.firstweekday()
        calendar.setfirstweekday(calendar.SUNDAY)
        self.assertEqual(calendar.firstweekday(), calendar.SUNDAY)
        calendar.setfirstweekday(calendar.MONDAY)
        self.assertEqual(calendar.firstweekday(), calendar.MONDAY)
        calendar.setfirstweekday(orig)
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
446 447 448 449 450 
    def test_illegal_weekday_reported(self):
        with self.assertRaisesRegex(calendar.IllegalWeekdayError, '123'):
            calendar.setfirstweekday(123)

    def test_enumerate_weekdays(self):
Skip Montanaro's avatar
added test case to catch index errors with _localized_name class
693c6c44
 
Skip Montanaro committed
451 452 453 
        self.assertRaises(IndexError, calendar.day_abbr.__getitem__, -10)
        self.assertRaises(IndexError, calendar.day_name.__getitem__, 10)
        self.assertEqual(len([d for d in calendar.day_abbr]), 7)
Fred Drake's avatar
Very small test suite for the calendar module, mostly to check a constraint
d077ca1e
 
Fred Drake committed
454
Tim Peters's avatar
SF bug 533234: tm_isdst > 1 Passed to strftime.
0c2c8e77
 
Tim Peters committed
455 456 457 458 459 460 
    def test_days(self):
        for attr in "day_name", "day_abbr":
            value = getattr(calendar, attr)
            self.assertEqual(len(value), 7)
            self.assertEqual(len(value[:]), 7)
            # ensure they're all unique
Tim Peters's avatar
SF bug 1065388: calendar day/month name lookup too slow
bbc0d440
 
Tim Peters committed
461 462 463 
            self.assertEqual(len(set(value)), 7)
            # verify it "acts like a sequence" in two forms of iteration
            self.assertEqual(value[::-1], list(reversed(value)))
Tim Peters's avatar
SF bug 533234: tm_isdst > 1 Passed to strftime.
0c2c8e77
 
Tim Peters committed
464 465 466 467 468 469 470 471 

    def test_months(self):
        for attr in "month_name", "month_abbr":
            value = getattr(calendar, attr)
            self.assertEqual(len(value), 13)
            self.assertEqual(len(value[:]), 13)
            self.assertEqual(value[0], "")
            # ensure they're all unique
Tim Peters's avatar
SF bug 1065388: calendar day/month name lookup too slow
bbc0d440
 
Tim Peters committed
472 473 474 
            self.assertEqual(len(set(value)), 13)
            # verify it "acts like a sequence" in two forms of iteration
            self.assertEqual(value[::-1], list(reversed(value)))
Tim Peters's avatar
SF bug 533234: tm_isdst > 1 Passed to strftime.
0c2c8e77
 
Tim Peters committed
475
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
476 
    def test_locale_calendars(self):
Georg Brandl's avatar
#10092: Properly reset locale in Locale*Calendar classes. The context manager…
7004bd1a
 
Georg Brandl committed
477 478 
        # ensure that Locale{Text,HTML}Calendar resets the locale properly
        # (it is still not thread-safe though)
Georg Brandl's avatar
Fix r85728: use "" to mean the system default locale, which should work on more systems.
9450cc05
 
Georg Brandl committed
479 
        old_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)
Georg Brandl's avatar
#10092: Properly reset locale in Locale*Calendar classes. The context manager…
7004bd1a
 
Georg Brandl committed
480 
        try:
Serhiy Storchaka's avatar
Added test to ensure localized calendar methods return strings and not bytes.
f9fcdb7e
 
Serhiy Storchaka committed
481 482 483 
            cal = calendar.LocaleTextCalendar(locale='')
            local_weekday = cal.formatweekday(1, 10)
            local_month = cal.formatmonthname(2010, 10, 10)
Georg Brandl's avatar
#10092: Properly reset locale in Locale*Calendar classes. The context manager…
7004bd1a
 
Georg Brandl committed
484 
        except locale.Error:
Georg Brandl's avatar
Fix r85728: use "" to mean the system default locale, which should work on more systems.
9450cc05
 
Georg Brandl committed
485 
            # cannot set the system default locale -- skip rest of test
Serhiy Storchaka's avatar
Added test to ensure localized calendar methods return strings and not bytes.
f9fcdb7e
 
Serhiy Storchaka committed
486 487 488 489 490 491 492 493 494 495 
            raise unittest.SkipTest('cannot set the system default locale')
        self.assertIsInstance(local_weekday, str)
        self.assertIsInstance(local_month, str)
        self.assertEqual(len(local_weekday), 10)
        self.assertGreaterEqual(len(local_month), 10)
        cal = calendar.LocaleHTMLCalendar(locale='')
        local_weekday = cal.formatweekday(1)
        local_month = cal.formatmonthname(2010, 10)
        self.assertIsInstance(local_weekday, str)
        self.assertIsInstance(local_month, str)
Georg Brandl's avatar
#10092: Properly reset locale in Locale*Calendar classes. The context manager…
7004bd1a
 
Georg Brandl committed
496 
        new_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)
Ezio Melotti's avatar
#9424: Replace deprecated assert* methods in the Python test suite.
b3aedd48
 
Ezio Melotti committed
497 
        self.assertEqual(old_october, new_october)
Georg Brandl's avatar
#10092: Properly reset locale in Locale*Calendar classes. The context manager…
7004bd1a
 
Georg Brandl committed
498
Ezio Melotti's avatar
#15421: fix an OverflowError in Calendar.itermonthdates() after…
85710a40
 
Ezio Melotti committed
499 500 501 502 503 
    def test_itermonthdates(self):
        # ensure itermonthdates doesn't overflow after datetime.MAXYEAR
        # see #15421
        list(calendar.Calendar().itermonthdates(datetime.MAXYEAR, 12))
Alexander Belopolsky's avatar
Issue #28253: Fixed calendar functions for extreme months: 0001-01 and 9999-12.
957b7569
 
Alexander Belopolsky committed
504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 
    def test_itermonthdays(self):
        for firstweekday in range(7):
            cal = calendar.Calendar(firstweekday)
            # Test the extremes, see #28253 and #26650
            for y, m in [(1, 1), (9999, 12)]:
                days = list(cal.itermonthdays(y, m))
                self.assertIn(len(days), (35, 42))
        # Test a short month
        cal = calendar.Calendar(firstweekday=3)
        days = list(cal.itermonthdays(2001, 2))
        self.assertEqual(days, list(range(1, 29)))

    def test_itermonthdays2(self):
        for firstweekday in range(7):
            cal = calendar.Calendar(firstweekday)
            # Test the extremes, see #28253 and #26650
            for y, m in [(1, 1), (9999, 12)]:
                days = list(cal.itermonthdays2(y, m))
                self.assertEqual(days[0][1], firstweekday)
                self.assertEqual(days[-1][1], (firstweekday - 1) % 7)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
525 526 527 528 529 530 531 532 533 534 535 536 

class MonthCalendarTestCase(unittest.TestCase):
    def setUp(self):
        self.oldfirstweekday = calendar.firstweekday()
        calendar.setfirstweekday(self.firstweekday)

    def tearDown(self):
        calendar.setfirstweekday(self.oldfirstweekday)

    def check_weeks(self, year, month, weeks):
        cal = calendar.monthcalendar(year, month)
        self.assertEqual(len(cal), len(weeks))
Guido van Rossum's avatar
Merged revisions 55007-55179 via svnmerge from
805365ee
 
Guido van Rossum committed
537 
        for i in range(len(weeks)):
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
538 539 540 541 542 543 544 
            self.assertEqual(weeks[i], sum(day != 0 for day in cal[i]))


class MondayTestCase(MonthCalendarTestCase):
    firstweekday = calendar.MONDAY

    def test_february(self):
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
545 
        # A 28-day february starting on monday (7+7+7+7 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
546 547 
        self.check_weeks(1999, 2, (7, 7, 7, 7))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
548 
        # A 28-day february starting on tuesday (6+7+7+7+1 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
549 550 
        self.check_weeks(2005, 2, (6, 7, 7, 7, 1))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
551 
        # A 28-day february starting on sunday (1+7+7+7+6 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
552 553 
        self.check_weeks(1987, 2, (1, 7, 7, 7, 6))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
554 
        # A 29-day february starting on monday (7+7+7+7+1 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
555 556 
        self.check_weeks(1988, 2, (7, 7, 7, 7, 1))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
557 
        # A 29-day february starting on tuesday (6+7+7+7+2 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
558 559 
        self.check_weeks(1972, 2, (6, 7, 7, 7, 2))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
560 
        # A 29-day february starting on sunday (1+7+7+7+7 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
561 562 563 
        self.check_weeks(2004, 2, (1, 7, 7, 7, 7))

    def test_april(self):
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
564 
        # A 30-day april starting on monday (7+7+7+7+2 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
565 566 
        self.check_weeks(1935, 4, (7, 7, 7, 7, 2))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
567 
        # A 30-day april starting on tuesday (6+7+7+7+3 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
568 569 
        self.check_weeks(1975, 4, (6, 7, 7, 7, 3))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
570 
        # A 30-day april starting on sunday (1+7+7+7+7+1 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
571 572 
        self.check_weeks(1945, 4, (1, 7, 7, 7, 7, 1))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
573 
        # A 30-day april starting on saturday (2+7+7+7+7 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
574 575 
        self.check_weeks(1995, 4, (2, 7, 7, 7, 7))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
576 
        # A 30-day april starting on friday (3+7+7+7+6 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
577 578 579 
        self.check_weeks(1994, 4, (3, 7, 7, 7, 6))

    def test_december(self):
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
580 
        # A 31-day december starting on monday (7+7+7+7+3 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
581 582 
        self.check_weeks(1980, 12, (7, 7, 7, 7, 3))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
583 
        # A 31-day december starting on tuesday (6+7+7+7+4 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
584 585 
        self.check_weeks(1987, 12, (6, 7, 7, 7, 4))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
586 
        # A 31-day december starting on sunday (1+7+7+7+7+2 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
587 588 
        self.check_weeks(1968, 12, (1, 7, 7, 7, 7, 2))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
589 
        # A 31-day december starting on thursday (4+7+7+7+6 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
590 591 
        self.check_weeks(1988, 12, (4, 7, 7, 7, 6))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
592 
        # A 31-day december starting on friday (3+7+7+7+7 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
593 594 
        self.check_weeks(2017, 12, (3, 7, 7, 7, 7))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
595 
        # A 31-day december starting on saturday (2+7+7+7+7+1 days)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
596 597 598 
        self.check_weeks(2068, 12, (2, 7, 7, 7, 7, 1))
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
599 600 601 602 
class SundayTestCase(MonthCalendarTestCase):
    firstweekday = calendar.SUNDAY

    def test_february(self):
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
603 
        # A 28-day february starting on sunday (7+7+7+7 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
604 605 
        self.check_weeks(2009, 2, (7, 7, 7, 7))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
606 
        # A 28-day february starting on monday (6+7+7+7+1 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
607 608 
        self.check_weeks(1999, 2, (6, 7, 7, 7, 1))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
609 
        # A 28-day february starting on saturday (1+7+7+7+6 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
610 611 
        self.check_weeks(1997, 2, (1, 7, 7, 7, 6))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
612 
        # A 29-day february starting on sunday (7+7+7+7+1 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
613 614 
        self.check_weeks(2004, 2, (7, 7, 7, 7, 1))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
615 
        # A 29-day february starting on monday (6+7+7+7+2 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
616 617 
        self.check_weeks(1960, 2, (6, 7, 7, 7, 2))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
618 
        # A 29-day february starting on saturday (1+7+7+7+7 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
619 620 621 
        self.check_weeks(1964, 2, (1, 7, 7, 7, 7))

    def test_april(self):
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
622 
        # A 30-day april starting on sunday (7+7+7+7+2 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
623 624 
        self.check_weeks(1923, 4, (7, 7, 7, 7, 2))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
625 
        # A 30-day april starting on monday (6+7+7+7+3 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
626 627 
        self.check_weeks(1918, 4, (6, 7, 7, 7, 3))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
628 
        # A 30-day april starting on saturday (1+7+7+7+7+1 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
629 630 
        self.check_weeks(1950, 4, (1, 7, 7, 7, 7, 1))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
631 
        # A 30-day april starting on friday (2+7+7+7+7 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
632 633 
        self.check_weeks(1960, 4, (2, 7, 7, 7, 7))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
634 
        # A 30-day april starting on thursday (3+7+7+7+6 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
635 636 637 
        self.check_weeks(1909, 4, (3, 7, 7, 7, 6))

    def test_december(self):
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
638 
        # A 31-day december starting on sunday (7+7+7+7+3 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
639 640 
        self.check_weeks(2080, 12, (7, 7, 7, 7, 3))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
641 
        # A 31-day december starting on monday (6+7+7+7+4 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
642 643 
        self.check_weeks(1941, 12, (6, 7, 7, 7, 4))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
644 
        # A 31-day december starting on saturday (1+7+7+7+7+2 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
645 646 
        self.check_weeks(1923, 12, (1, 7, 7, 7, 7, 2))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
647 
        # A 31-day december starting on wednesday (4+7+7+7+6 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
648 649 
        self.check_weeks(1948, 12, (4, 7, 7, 7, 6))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
650 
        # A 31-day december starting on thursday (3+7+7+7+7 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
651 652 
        self.check_weeks(1927, 12, (3, 7, 7, 7, 7))
Thomas Wouters's avatar
Merge p3yk branch with the trunk up to revision 45595. This breaks a fair
49fd7fa4
 
Thomas Wouters committed
653 
        # A 31-day december starting on friday (2+7+7+7+7+1 days)
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
654 655 
        self.check_weeks(1995, 12, (2, 7, 7, 7, 7, 1))
Alexander Belopolsky's avatar
Issue 6280: Tests and simpler implementation for calendar.timegm
43ca710a
 
Alexander Belopolsky committed
656 657 658 659 660 661 662 
class TimegmTestCase(unittest.TestCase):
    TIMESTAMPS = [0, 10, 100, 1000, 10000, 100000, 1000000,
                  1234567890, 1262304000, 1275785153,]
    def test_timegm(self):
        for secs in self.TIMESTAMPS:
            tuple = time.gmtime(secs)
            self.assertEqual(secs, calendar.timegm(tuple))
Walter Dörwald's avatar
Add another bunch of test cases for calendars with Sunday as the
302fa6dc
 
Walter Dörwald committed
663
Brett Cannon's avatar
Test calendar.monthrange.
cfae36fa
 
Brett Cannon committed
664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 
class MonthRangeTestCase(unittest.TestCase):
    def test_january(self):
        # Tests valid lower boundary case.
        self.assertEqual(calendar.monthrange(2004,1), (3,31))

    def test_february_leap(self):
        # Tests February during leap year.
        self.assertEqual(calendar.monthrange(2004,2), (6,29))

    def test_february_nonleap(self):
        # Tests February in non-leap year.
        self.assertEqual(calendar.monthrange(2010,2), (0,28))

    def test_december(self):
        # Tests valid upper boundary case.
        self.assertEqual(calendar.monthrange(2004,12), (2,31))

    def test_zeroth_month(self):
        # Tests low invalid boundary case.
        with self.assertRaises(calendar.IllegalMonthError):
            calendar.monthrange(2004, 0)

    def test_thirteenth_month(self):
        # Tests high invalid boundary case.
        with self.assertRaises(calendar.IllegalMonthError):
            calendar.monthrange(2004, 13)
R David Murray's avatar
#14796: improve calendar test coverage.
d0c86956
 
R David Murray committed
691 692 693 694 
    def test_illegal_month_reported(self):
        with self.assertRaisesRegex(calendar.IllegalMonthError, '65'):
            calendar.monthrange(2004, 65)
R. David Murray's avatar
#9361: add some tests for calendar.leapdays
1daaf9e6
 
R. David Murray committed
695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 
class LeapdaysTestCase(unittest.TestCase):
    def test_no_range(self):
        # test when no range i.e. two identical years as args
        self.assertEqual(calendar.leapdays(2010,2010), 0)

    def test_no_leapdays(self):
        # test when no leap years in range
        self.assertEqual(calendar.leapdays(2010,2011), 0)

    def test_no_leapdays_upper_boundary(self):
        # test no leap years in range, when upper boundary is a leap year
        self.assertEqual(calendar.leapdays(2010,2012), 0)

    def test_one_leapday_lower_boundary(self):
        # test when one leap year in range, lower boundary is leap year
        self.assertEqual(calendar.leapdays(2012,2013), 1)

    def test_several_leapyears_in_range(self):
        self.assertEqual(calendar.leapdays(1997,2020), 5)
Brett Cannon's avatar
Test calendar.monthrange.
cfae36fa
 
Brett Cannon committed
715
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
716 717 718 719 720 721 722 723 724 
def conv(s):
    return s.replace('\n', os.linesep).encode()

class CommandLineTestCase(unittest.TestCase):
    def run_ok(self, *args):
        return assert_python_ok('-m', 'calendar', *args)[1]

    def assertFailure(self, *args):
        rc, stdout, stderr = assert_python_failure('-m', 'calendar', *args)
Serhiy Storchaka's avatar
Issue #18973: Command-line interface of the calendar module now uses argparse
9785261d
 
Serhiy Storchaka committed
725 
        self.assertIn(b'usage:', stderr)
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
726 727 728 729 
        self.assertEqual(rc, 2)

    def test_help(self):
        stdout = self.run_ok('-h')
Serhiy Storchaka's avatar
Issue #18973: Command-line interface of the calendar module now uses argparse
9785261d
 
Serhiy Storchaka committed
730 
        self.assertIn(b'usage:', stdout)
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
731 732 733 734 735 
        self.assertIn(b'calendar.py', stdout)
        self.assertIn(b'--help', stdout)

    def test_illegal_arguments(self):
        self.assertFailure('-z')
Serhiy Storchaka's avatar
Issue #18973: Command-line interface of the calendar module now uses argparse
9785261d
 
Serhiy Storchaka committed
736 737 
        self.assertFailure('spam')
        self.assertFailure('2004', 'spam')
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 
        self.assertFailure('-t', 'html', '2004', '1')

    def test_output_current_year(self):
        stdout = self.run_ok()
        year = datetime.datetime.now().year
        self.assertIn((' %s' % year).encode(), stdout)
        self.assertIn(b'January', stdout)
        self.assertIn(b'Mo Tu We Th Fr Sa Su', stdout)

    def test_output_year(self):
        stdout = self.run_ok('2004')
        self.assertEqual(stdout, conv(result_2004_text))

    def test_output_month(self):
        stdout = self.run_ok('2004', '1')
        self.assertEqual(stdout, conv(result_2004_01_text))

    def test_option_encoding(self):
        self.assertFailure('-e')
        self.assertFailure('--encoding')
        stdout = self.run_ok('--encoding', 'utf-16-le', '2004')
        self.assertEqual(stdout, result_2004_text.encode('utf-16-le'))

    def test_option_locale(self):
        self.assertFailure('-L')
        self.assertFailure('--locale')
        self.assertFailure('-L', 'en')
        lang, enc = locale.getdefaultlocale()
        lang = lang or 'C'
        enc = enc or 'UTF-8'
        try:
            oldlocale = locale.getlocale(locale.LC_TIME)
            try:
                locale.setlocale(locale.LC_TIME, (lang, enc))
            finally:
                locale.setlocale(locale.LC_TIME, oldlocale)
        except (locale.Error, ValueError):
            self.skipTest('cannot set the system default locale')
        stdout = self.run_ok('--locale', lang, '--encoding', enc, '2004')
        self.assertIn('2004'.encode(enc), stdout)

    def test_option_width(self):
        self.assertFailure('-w')
        self.assertFailure('--width')
        self.assertFailure('-w', 'spam')
        stdout = self.run_ok('--width', '3', '2004')
        self.assertIn(b'Mon Tue Wed Thu Fri Sat Sun', stdout)

    def test_option_lines(self):
        self.assertFailure('-l')
        self.assertFailure('--lines')
        self.assertFailure('-l', 'spam')
        stdout = self.run_ok('--lines', '2', '2004')
        self.assertIn(conv('December\n\nMo Tu We'), stdout)

    def test_option_spacing(self):
        self.assertFailure('-s')
        self.assertFailure('--spacing')
        self.assertFailure('-s', 'spam')
        stdout = self.run_ok('--spacing', '8', '2004')
        self.assertIn(b'Su        Mo', stdout)

    def test_option_months(self):
        self.assertFailure('-m')
        self.assertFailure('--month')
        self.assertFailure('-m', 'spam')
        stdout = self.run_ok('--months', '1', '2004')
        self.assertIn(conv('\nMo Tu We Th Fr Sa Su\n'), stdout)

    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, conv(result_2004_text))
        stdout = self.run_ok('--type', 'html', '2004')
Senthil Kumaran's avatar
Fix closes Issue10087 - fixing the output of calendar display in the html…
962fed91
 
Senthil Kumaran committed
814 
        self.assertEqual(stdout[:6], b'<?xml ')
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
815 816 817 818 819 820 821 822 823 824 825 826 827 828 
        self.assertIn(b'<title>Calendar for 2004</title>', stdout)

    def test_html_output_current_year(self):
        stdout = self.run_ok('--type', 'html')
        year = datetime.datetime.now().year
        self.assertIn(('<title>Calendar for %s</title>' % year).encode(),
                      stdout)
        self.assertIn(b'<tr><th colspan="7" class="month">January</th></tr>',
                      stdout)

    def test_html_output_year_encoding(self):
        stdout = self.run_ok('-t', 'html', '--encoding', 'ascii', '2004')
        self.assertEqual(stdout,
                         (result_2004_html % {'e': 'ascii'}).encode('ascii'))
Senthil Kumaran's avatar
Fix closes Issue10087 - fixing the output of calendar display in the html…
962fed91
 
Senthil Kumaran committed
829
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
830 831 832 833 834 835 
    def test_html_output_year_css(self):
        self.assertFailure('-t', 'html', '-c')
        self.assertFailure('-t', 'html', '--css')
        stdout = self.run_ok('-t', 'html', '--css', 'custom.css', '2004')
        self.assertIn(b'<link rel="stylesheet" type="text/css" '
                      b'href="custom.css" />', stdout)
Walter Dörwald's avatar
Add tests that check the result of calendar.monthcalendar() for a set
59ad4568
 
Walter Dörwald committed
836
Fred Drake's avatar
Very small test suite for the calendar module, mostly to check a constraint
d077ca1e
 
Fred Drake committed
837
Martin Panter's avatar
Issue #23883: Add missing APIs to calendar.__all__
4eb376c4
 
Martin Panter committed
838 839 
class MiscTestCase(unittest.TestCase):
    def test__all__(self):
Serhiy Storchaka's avatar
Issue #23883: Removed redundant names from blacklists.
2c1d3e32
 
Serhiy Storchaka committed
840 
        blacklist = {'mdays', 'January', 'February', 'EPOCH',
Martin Panter's avatar
Issue #23883: Add missing APIs to calendar.__all__
4eb376c4
 
Martin Panter committed
841 842 843 844 845 846 
                     'MONDAY', 'TUESDAY', 'WEDNESDAY', 'THURSDAY', 'FRIDAY',
                     'SATURDAY', 'SUNDAY', 'different_locale', 'c',
                     'prweek', 'week', 'format', 'formatstring', 'main'}
        support.check__all__(self, calendar, blacklist=blacklist)
Fred Drake's avatar
Very small test suite for the calendar module, mostly to check a constraint
d077ca1e
 
Fred Drake committed
847 
if __name__ == "__main__":
Serhiy Storchaka's avatar
Issue #18982: Add tests for CLI of the calendar module.
aef859c5
 
Serhiy Storchaka committed
848 
    unittest.main()