test_calendar.py 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609
  1. import calendar
  2. import unittest
  3. from test import test_support
  4. from test.script_helper import assert_python_ok, assert_python_failure
  5. import locale
  6. import datetime
  7. import os
  8. result_2004_01_text = """\
  9. January 2004
  10. Mo Tu We Th Fr Sa Su
  11. 1 2 3 4
  12. 5 6 7 8 9 10 11
  13. 12 13 14 15 16 17 18
  14. 19 20 21 22 23 24 25
  15. 26 27 28 29 30 31
  16. """
  17. result_2004_text = """\
  18. 2004
  19. January February March
  20. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  21. 1 2 3 4 1 1 2 3 4 5 6 7
  22. 5 6 7 8 9 10 11 2 3 4 5 6 7 8 8 9 10 11 12 13 14
  23. 12 13 14 15 16 17 18 9 10 11 12 13 14 15 15 16 17 18 19 20 21
  24. 19 20 21 22 23 24 25 16 17 18 19 20 21 22 22 23 24 25 26 27 28
  25. 26 27 28 29 30 31 23 24 25 26 27 28 29 29 30 31
  26. April May June
  27. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  28. 1 2 3 4 1 2 1 2 3 4 5 6
  29. 5 6 7 8 9 10 11 3 4 5 6 7 8 9 7 8 9 10 11 12 13
  30. 12 13 14 15 16 17 18 10 11 12 13 14 15 16 14 15 16 17 18 19 20
  31. 19 20 21 22 23 24 25 17 18 19 20 21 22 23 21 22 23 24 25 26 27
  32. 26 27 28 29 30 24 25 26 27 28 29 30 28 29 30
  33. 31
  34. July August September
  35. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  36. 1 2 3 4 1 1 2 3 4 5
  37. 5 6 7 8 9 10 11 2 3 4 5 6 7 8 6 7 8 9 10 11 12
  38. 12 13 14 15 16 17 18 9 10 11 12 13 14 15 13 14 15 16 17 18 19
  39. 19 20 21 22 23 24 25 16 17 18 19 20 21 22 20 21 22 23 24 25 26
  40. 26 27 28 29 30 31 23 24 25 26 27 28 29 27 28 29 30
  41. 30 31
  42. October November December
  43. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  44. 1 2 3 1 2 3 4 5 6 7 1 2 3 4 5
  45. 4 5 6 7 8 9 10 8 9 10 11 12 13 14 6 7 8 9 10 11 12
  46. 11 12 13 14 15 16 17 15 16 17 18 19 20 21 13 14 15 16 17 18 19
  47. 18 19 20 21 22 23 24 22 23 24 25 26 27 28 20 21 22 23 24 25 26
  48. 25 26 27 28 29 30 31 29 30 27 28 29 30 31
  49. """
  50. result_2004_html = """\
  51. <?xml version="1.0" encoding="ascii"?>
  52. <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
  53. <html>
  54. <head>
  55. <meta http-equiv="Content-Type" content="text/html; charset=ascii" />
  56. <link rel="stylesheet" type="text/css" href="calendar.css" />
  57. <title>Calendar for 2004</title>
  58. </head>
  59. <body>
  60. <table border="0" cellpadding="0" cellspacing="0" class="year">
  61. <tr><th colspan="3" class="year">2004</th></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  62. <tr><th colspan="7" class="month">January</th></tr>
  63. <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>
  64. <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>
  65. <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>
  66. <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>
  67. <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>
  68. <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>
  69. </table>
  70. </td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  71. <tr><th colspan="7" class="month">February</th></tr>
  72. <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>
  73. <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>
  74. <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>
  75. <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>
  76. <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>
  77. <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>
  78. </table>
  79. </td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  80. <tr><th colspan="7" class="month">March</th></tr>
  81. <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>
  82. <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>
  83. <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>
  84. <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>
  85. <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>
  86. <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>
  87. </table>
  88. </td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  89. <tr><th colspan="7" class="month">April</th></tr>
  90. <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>
  91. <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>
  92. <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>
  93. <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>
  94. <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>
  95. <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>
  96. </table>
  97. </td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  98. <tr><th colspan="7" class="month">May</th></tr>
  99. <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>
  100. <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>
  101. <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>
  102. <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>
  103. <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>
  104. <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>
  105. <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>
  106. </table>
  107. </td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  108. <tr><th colspan="7" class="month">June</th></tr>
  109. <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>
  110. <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>
  111. <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>
  112. <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>
  113. <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>
  114. <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>
  115. </table>
  116. </td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  117. <tr><th colspan="7" class="month">July</th></tr>
  118. <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>
  119. <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>
  120. <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>
  121. <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>
  122. <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>
  123. <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>
  124. </table>
  125. </td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  126. <tr><th colspan="7" class="month">August</th></tr>
  127. <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>
  128. <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>
  129. <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>
  130. <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>
  131. <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>
  132. <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>
  133. <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>
  134. </table>
  135. </td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  136. <tr><th colspan="7" class="month">September</th></tr>
  137. <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>
  138. <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>
  139. <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>
  140. <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>
  141. <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>
  142. <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>
  143. </table>
  144. </td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  145. <tr><th colspan="7" class="month">October</th></tr>
  146. <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>
  147. <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>
  148. <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>
  149. <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>
  150. <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>
  151. <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>
  152. </table>
  153. </td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  154. <tr><th colspan="7" class="month">November</th></tr>
  155. <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>
  156. <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>
  157. <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>
  158. <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>
  159. <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>
  160. <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>
  161. </table>
  162. </td><td><table border="0" cellpadding="0" cellspacing="0" class="month">
  163. <tr><th colspan="7" class="month">December</th></tr>
  164. <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>
  165. <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>
  166. <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>
  167. <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>
  168. <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>
  169. <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>
  170. </table>
  171. </td></tr></table></body>
  172. </html>
  173. """
  174. class OutputTestCase(unittest.TestCase):
  175. def normalize_calendar(self, s):
  176. # Filters out locale dependent strings
  177. def neitherspacenordigit(c):
  178. return not c.isspace() and not c.isdigit()
  179. lines = []
  180. for line in s.splitlines(False):
  181. # Drop texts, as they are locale dependent
  182. if line and not filter(neitherspacenordigit, line):
  183. lines.append(line)
  184. return lines
  185. def test_output(self):
  186. self.assertEqual(
  187. self.normalize_calendar(calendar.calendar(2004)),
  188. self.normalize_calendar(result_2004_text)
  189. )
  190. def test_output_textcalendar(self):
  191. self.assertEqual(
  192. calendar.TextCalendar().formatyear(2004).strip(),
  193. result_2004_text.strip()
  194. )
  195. def test_output_htmlcalendar(self):
  196. self.assertEqual(
  197. calendar.HTMLCalendar().formatyearpage(2004).strip(),
  198. result_2004_html.strip()
  199. )
  200. class CalendarTestCase(unittest.TestCase):
  201. def test_isleap(self):
  202. # Make sure that the return is right for a few years, and
  203. # ensure that the return values are 1 or 0, not just true or
  204. # false (see SF bug #485794). Specific additional tests may
  205. # be appropriate; this tests a single "cycle".
  206. self.assertEqual(calendar.isleap(2000), 1)
  207. self.assertEqual(calendar.isleap(2001), 0)
  208. self.assertEqual(calendar.isleap(2002), 0)
  209. self.assertEqual(calendar.isleap(2003), 0)
  210. def test_setfirstweekday(self):
  211. self.assertRaises(ValueError, calendar.setfirstweekday, 'flabber')
  212. self.assertRaises(ValueError, calendar.setfirstweekday, -1)
  213. self.assertRaises(ValueError, calendar.setfirstweekday, 200)
  214. orig = calendar.firstweekday()
  215. calendar.setfirstweekday(calendar.SUNDAY)
  216. self.assertEqual(calendar.firstweekday(), calendar.SUNDAY)
  217. calendar.setfirstweekday(calendar.MONDAY)
  218. self.assertEqual(calendar.firstweekday(), calendar.MONDAY)
  219. calendar.setfirstweekday(orig)
  220. def test_enumerateweekdays(self):
  221. self.assertRaises(IndexError, calendar.day_abbr.__getitem__, -10)
  222. self.assertRaises(IndexError, calendar.day_name.__getitem__, 10)
  223. self.assertEqual(len([d for d in calendar.day_abbr]), 7)
  224. def test_days(self):
  225. for attr in "day_name", "day_abbr":
  226. value = getattr(calendar, attr)
  227. self.assertEqual(len(value), 7)
  228. self.assertEqual(len(value[:]), 7)
  229. # ensure they're all unique
  230. self.assertEqual(len(set(value)), 7)
  231. # verify it "acts like a sequence" in two forms of iteration
  232. self.assertEqual(value[::-1], list(reversed(value)))
  233. def test_months(self):
  234. for attr in "month_name", "month_abbr":
  235. value = getattr(calendar, attr)
  236. self.assertEqual(len(value), 13)
  237. self.assertEqual(len(value[:]), 13)
  238. self.assertEqual(value[0], "")
  239. # ensure they're all unique
  240. self.assertEqual(len(set(value)), 13)
  241. # verify it "acts like a sequence" in two forms of iteration
  242. self.assertEqual(value[::-1], list(reversed(value)))
  243. def test_localecalendars(self):
  244. # ensure that Locale{Text,HTML}Calendar resets the locale properly
  245. # (it is still not thread-safe though)
  246. old_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)
  247. try:
  248. cal = calendar.LocaleTextCalendar(locale='')
  249. local_weekday = cal.formatweekday(1, 10)
  250. local_month = cal.formatmonthname(2010, 10, 10)
  251. except locale.Error:
  252. # cannot set the system default locale -- skip rest of test
  253. raise unittest.SkipTest('cannot set the system default locale')
  254. # should be encodable
  255. local_weekday.encode('utf-8')
  256. local_month.encode('utf-8')
  257. self.assertEqual(len(local_weekday), 10)
  258. self.assertGreaterEqual(len(local_month), 10)
  259. cal = calendar.LocaleHTMLCalendar(locale='')
  260. local_weekday = cal.formatweekday(1)
  261. local_month = cal.formatmonthname(2010, 10)
  262. # should be encodable
  263. local_weekday.encode('utf-8')
  264. local_month.encode('utf-8')
  265. new_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)
  266. self.assertEqual(old_october, new_october)
  267. def test_itermonthdates(self):
  268. # ensure itermonthdates doesn't overflow after datetime.MAXYEAR
  269. # see #15421
  270. list(calendar.Calendar().itermonthdates(datetime.MAXYEAR, 12))
  271. class MonthCalendarTestCase(unittest.TestCase):
  272. def setUp(self):
  273. self.oldfirstweekday = calendar.firstweekday()
  274. calendar.setfirstweekday(self.firstweekday)
  275. def tearDown(self):
  276. calendar.setfirstweekday(self.oldfirstweekday)
  277. def check_weeks(self, year, month, weeks):
  278. cal = calendar.monthcalendar(year, month)
  279. self.assertEqual(len(cal), len(weeks))
  280. for i in xrange(len(weeks)):
  281. self.assertEqual(weeks[i], sum(day != 0 for day in cal[i]))
  282. class MondayTestCase(MonthCalendarTestCase):
  283. firstweekday = calendar.MONDAY
  284. def test_february(self):
  285. # A 28-day february starting on monday (7+7+7+7 days)
  286. self.check_weeks(1999, 2, (7, 7, 7, 7))
  287. # A 28-day february starting on tuesday (6+7+7+7+1 days)
  288. self.check_weeks(2005, 2, (6, 7, 7, 7, 1))
  289. # A 28-day february starting on sunday (1+7+7+7+6 days)
  290. self.check_weeks(1987, 2, (1, 7, 7, 7, 6))
  291. # A 29-day february starting on monday (7+7+7+7+1 days)
  292. self.check_weeks(1988, 2, (7, 7, 7, 7, 1))
  293. # A 29-day february starting on tuesday (6+7+7+7+2 days)
  294. self.check_weeks(1972, 2, (6, 7, 7, 7, 2))
  295. # A 29-day february starting on sunday (1+7+7+7+7 days)
  296. self.check_weeks(2004, 2, (1, 7, 7, 7, 7))
  297. def test_april(self):
  298. # A 30-day april starting on monday (7+7+7+7+2 days)
  299. self.check_weeks(1935, 4, (7, 7, 7, 7, 2))
  300. # A 30-day april starting on tuesday (6+7+7+7+3 days)
  301. self.check_weeks(1975, 4, (6, 7, 7, 7, 3))
  302. # A 30-day april starting on sunday (1+7+7+7+7+1 days)
  303. self.check_weeks(1945, 4, (1, 7, 7, 7, 7, 1))
  304. # A 30-day april starting on saturday (2+7+7+7+7 days)
  305. self.check_weeks(1995, 4, (2, 7, 7, 7, 7))
  306. # A 30-day april starting on friday (3+7+7+7+6 days)
  307. self.check_weeks(1994, 4, (3, 7, 7, 7, 6))
  308. def test_december(self):
  309. # A 31-day december starting on monday (7+7+7+7+3 days)
  310. self.check_weeks(1980, 12, (7, 7, 7, 7, 3))
  311. # A 31-day december starting on tuesday (6+7+7+7+4 days)
  312. self.check_weeks(1987, 12, (6, 7, 7, 7, 4))
  313. # A 31-day december starting on sunday (1+7+7+7+7+2 days)
  314. self.check_weeks(1968, 12, (1, 7, 7, 7, 7, 2))
  315. # A 31-day december starting on thursday (4+7+7+7+6 days)
  316. self.check_weeks(1988, 12, (4, 7, 7, 7, 6))
  317. # A 31-day december starting on friday (3+7+7+7+7 days)
  318. self.check_weeks(2017, 12, (3, 7, 7, 7, 7))
  319. # A 31-day december starting on saturday (2+7+7+7+7+1 days)
  320. self.check_weeks(2068, 12, (2, 7, 7, 7, 7, 1))
  321. class SundayTestCase(MonthCalendarTestCase):
  322. firstweekday = calendar.SUNDAY
  323. def test_february(self):
  324. # A 28-day february starting on sunday (7+7+7+7 days)
  325. self.check_weeks(2009, 2, (7, 7, 7, 7))
  326. # A 28-day february starting on monday (6+7+7+7+1 days)
  327. self.check_weeks(1999, 2, (6, 7, 7, 7, 1))
  328. # A 28-day february starting on saturday (1+7+7+7+6 days)
  329. self.check_weeks(1997, 2, (1, 7, 7, 7, 6))
  330. # A 29-day february starting on sunday (7+7+7+7+1 days)
  331. self.check_weeks(2004, 2, (7, 7, 7, 7, 1))
  332. # A 29-day february starting on monday (6+7+7+7+2 days)
  333. self.check_weeks(1960, 2, (6, 7, 7, 7, 2))
  334. # A 29-day february starting on saturday (1+7+7+7+7 days)
  335. self.check_weeks(1964, 2, (1, 7, 7, 7, 7))
  336. def test_april(self):
  337. # A 30-day april starting on sunday (7+7+7+7+2 days)
  338. self.check_weeks(1923, 4, (7, 7, 7, 7, 2))
  339. # A 30-day april starting on monday (6+7+7+7+3 days)
  340. self.check_weeks(1918, 4, (6, 7, 7, 7, 3))
  341. # A 30-day april starting on saturday (1+7+7+7+7+1 days)
  342. self.check_weeks(1950, 4, (1, 7, 7, 7, 7, 1))
  343. # A 30-day april starting on friday (2+7+7+7+7 days)
  344. self.check_weeks(1960, 4, (2, 7, 7, 7, 7))
  345. # A 30-day april starting on thursday (3+7+7+7+6 days)
  346. self.check_weeks(1909, 4, (3, 7, 7, 7, 6))
  347. def test_december(self):
  348. # A 31-day december starting on sunday (7+7+7+7+3 days)
  349. self.check_weeks(2080, 12, (7, 7, 7, 7, 3))
  350. # A 31-day december starting on monday (6+7+7+7+4 days)
  351. self.check_weeks(1941, 12, (6, 7, 7, 7, 4))
  352. # A 31-day december starting on saturday (1+7+7+7+7+2 days)
  353. self.check_weeks(1923, 12, (1, 7, 7, 7, 7, 2))
  354. # A 31-day december starting on wednesday (4+7+7+7+6 days)
  355. self.check_weeks(1948, 12, (4, 7, 7, 7, 6))
  356. # A 31-day december starting on thursday (3+7+7+7+7 days)
  357. self.check_weeks(1927, 12, (3, 7, 7, 7, 7))
  358. # A 31-day december starting on friday (2+7+7+7+7+1 days)
  359. self.check_weeks(1995, 12, (2, 7, 7, 7, 7, 1))
  360. class MonthRangeTestCase(unittest.TestCase):
  361. def test_january(self):
  362. # Tests valid lower boundary case.
  363. self.assertEqual(calendar.monthrange(2004,1), (3,31))
  364. def test_february_leap(self):
  365. # Tests February during leap year.
  366. self.assertEqual(calendar.monthrange(2004,2), (6,29))
  367. def test_february_nonleap(self):
  368. # Tests February in non-leap year.
  369. self.assertEqual(calendar.monthrange(2010,2), (0,28))
  370. def test_december(self):
  371. # Tests valid upper boundary case.
  372. self.assertEqual(calendar.monthrange(2004,12), (2,31))
  373. def test_zeroth_month(self):
  374. # Tests low invalid boundary case.
  375. with self.assertRaises(calendar.IllegalMonthError):
  376. calendar.monthrange(2004, 0)
  377. def test_thirteenth_month(self):
  378. # Tests high invalid boundary case.
  379. with self.assertRaises(calendar.IllegalMonthError):
  380. calendar.monthrange(2004, 13)
  381. class LeapdaysTestCase(unittest.TestCase):
  382. def test_no_range(self):
  383. # test when no range i.e. two identical years as args
  384. self.assertEqual(calendar.leapdays(2010,2010), 0)
  385. def test_no_leapdays(self):
  386. # test when no leap years in range
  387. self.assertEqual(calendar.leapdays(2010,2011), 0)
  388. def test_no_leapdays_upper_boundary(self):
  389. # test no leap years in range, when upper boundary is a leap year
  390. self.assertEqual(calendar.leapdays(2010,2012), 0)
  391. def test_one_leapday_lower_boundary(self):
  392. # test when one leap year in range, lower boundary is leap year
  393. self.assertEqual(calendar.leapdays(2012,2013), 1)
  394. def test_several_leapyears_in_range(self):
  395. self.assertEqual(calendar.leapdays(1997,2020), 5)
  396. def conv(s):
  397. return s.replace('\n', os.linesep)
  398. class CommandLineTestCase(unittest.TestCase):
  399. def run_ok(self, *args):
  400. return assert_python_ok('-m', 'calendar', *args)[1]
  401. def assertFailure(self, *args):
  402. rc, stdout, stderr = assert_python_failure('-m', 'calendar', *args)
  403. self.assertIn(b'Usage:', stderr)
  404. self.assertEqual(rc, 2)
  405. def test_help(self):
  406. stdout = self.run_ok('-h')
  407. self.assertIn(b'Usage:', stdout)
  408. self.assertIn(b'calendar.py', stdout)
  409. self.assertIn(b'--help', stdout)
  410. def test_illegal_arguments(self):
  411. self.assertFailure('-z')
  412. #self.assertFailure('spam')
  413. #self.assertFailure('2004', 'spam')
  414. self.assertFailure('-t', 'html', '2004', '1')
  415. def test_output_current_year(self):
  416. stdout = self.run_ok()
  417. year = datetime.datetime.now().year
  418. self.assertIn((' %s' % year).encode(), stdout)
  419. self.assertIn(b'January', stdout)
  420. self.assertIn(b'Mo Tu We Th Fr Sa Su', stdout)
  421. def test_output_year(self):
  422. stdout = self.run_ok('2004')
  423. self.assertEqual(stdout.strip(), conv(result_2004_text).strip())
  424. def test_output_month(self):
  425. stdout = self.run_ok('2004', '1')
  426. self.assertEqual(stdout.strip(), conv(result_2004_01_text).strip())
  427. def test_option_encoding(self):
  428. self.assertFailure('-e')
  429. self.assertFailure('--encoding')
  430. stdout = self.run_ok('--encoding', 'utf-16-le', '2004')
  431. self.assertEqual(stdout.strip(), conv(result_2004_text.encode('utf-16-le')).strip())
  432. def test_option_locale(self):
  433. self.assertFailure('-L')
  434. self.assertFailure('--locale')
  435. self.assertFailure('-L', 'en')
  436. lang, enc = locale.getdefaultlocale()
  437. lang = lang or 'C'
  438. enc = enc or 'UTF-8'
  439. try:
  440. oldlocale = locale.getlocale(locale.LC_TIME)
  441. try:
  442. locale.setlocale(locale.LC_TIME, (lang, enc))
  443. finally:
  444. locale.setlocale(locale.LC_TIME, oldlocale)
  445. except (locale.Error, ValueError):
  446. self.skipTest('cannot set the system default locale')
  447. stdout = self.run_ok('--locale', lang, '--encoding', enc, '2004')
  448. self.assertIn('2004'.encode(enc), stdout)
  449. def test_option_width(self):
  450. self.assertFailure('-w')
  451. self.assertFailure('--width')
  452. self.assertFailure('-w', 'spam')
  453. stdout = self.run_ok('--width', '3', '2004')
  454. self.assertIn(b'Mon Tue Wed Thu Fri Sat Sun', stdout)
  455. def test_option_lines(self):
  456. self.assertFailure('-l')
  457. self.assertFailure('--lines')
  458. self.assertFailure('-l', 'spam')
  459. stdout = self.run_ok('--lines', '2', '2004')
  460. self.assertIn(conv('December\n\nMo Tu We'), stdout)
  461. def test_option_spacing(self):
  462. self.assertFailure('-s')
  463. self.assertFailure('--spacing')
  464. self.assertFailure('-s', 'spam')
  465. stdout = self.run_ok('--spacing', '8', '2004')
  466. self.assertIn(b'Su Mo', stdout)
  467. def test_option_months(self):
  468. self.assertFailure('-m')
  469. self.assertFailure('--month')
  470. self.assertFailure('-m', 'spam')
  471. stdout = self.run_ok('--months', '1', '2004')
  472. self.assertIn(conv('\nMo Tu We Th Fr Sa Su\n'), stdout)
  473. def test_option_type(self):
  474. self.assertFailure('-t')
  475. self.assertFailure('--type')
  476. self.assertFailure('-t', 'spam')
  477. stdout = self.run_ok('--type', 'text', '2004')
  478. self.assertEqual(stdout.strip(), conv(result_2004_text).strip())
  479. stdout = self.run_ok('--type', 'html', '2004')
  480. self.assertEqual(stdout[:6], b'<?xml ')
  481. self.assertIn(b'<title>Calendar for 2004</title>', stdout)
  482. def test_html_output_current_year(self):
  483. stdout = self.run_ok('--type', 'html')
  484. year = datetime.datetime.now().year
  485. self.assertIn(('<title>Calendar for %s</title>' % year).encode(),
  486. stdout)
  487. self.assertIn(b'<tr><th colspan="7" class="month">January</th></tr>',
  488. stdout)
  489. def test_html_output_year_encoding(self):
  490. stdout = self.run_ok('-t', 'html', '--encoding', 'ascii', '2004')
  491. self.assertEqual(stdout.strip(), conv(result_2004_html).strip())
  492. def test_html_output_year_css(self):
  493. self.assertFailure('-t', 'html', '-c')
  494. self.assertFailure('-t', 'html', '--css')
  495. stdout = self.run_ok('-t', 'html', '--css', 'custom.css', '2004')
  496. self.assertIn(b'<link rel="stylesheet" type="text/css" '
  497. b'href="custom.css" />', stdout)
  498. def test_main():
  499. test_support.run_unittest(
  500. OutputTestCase,
  501. CalendarTestCase,
  502. MondayTestCase,
  503. SundayTestCase,
  504. MonthRangeTestCase,
  505. LeapdaysTestCase,
  506. CommandLineTestCase,
  507. )
  508. if __name__ == "__main__":
  509. test_main()
  510. unittest.main()