test_calendar.py 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009
  1. import calendar
  2. import unittest
  3. from test import support
  4. from test.support.script_helper import assert_python_ok, assert_python_failure
  5. import time
  6. import locale
  7. import sys
  8. import datetime
  9. import os
  10. # From https://en.wikipedia.org/wiki/Leap_year_starting_on_Saturday
  11. result_0_02_text = """\
  12. February 0
  13. Mo Tu We Th Fr Sa Su
  14. 1 2 3 4 5 6
  15. 7 8 9 10 11 12 13
  16. 14 15 16 17 18 19 20
  17. 21 22 23 24 25 26 27
  18. 28 29
  19. """
  20. result_0_text = """\
  21. 0
  22. January February March
  23. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  24. 1 2 1 2 3 4 5 6 1 2 3 4 5
  25. 3 4 5 6 7 8 9 7 8 9 10 11 12 13 6 7 8 9 10 11 12
  26. 10 11 12 13 14 15 16 14 15 16 17 18 19 20 13 14 15 16 17 18 19
  27. 17 18 19 20 21 22 23 21 22 23 24 25 26 27 20 21 22 23 24 25 26
  28. 24 25 26 27 28 29 30 28 29 27 28 29 30 31
  29. 31
  30. April May June
  31. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  32. 1 2 1 2 3 4 5 6 7 1 2 3 4
  33. 3 4 5 6 7 8 9 8 9 10 11 12 13 14 5 6 7 8 9 10 11
  34. 10 11 12 13 14 15 16 15 16 17 18 19 20 21 12 13 14 15 16 17 18
  35. 17 18 19 20 21 22 23 22 23 24 25 26 27 28 19 20 21 22 23 24 25
  36. 24 25 26 27 28 29 30 29 30 31 26 27 28 29 30
  37. July August September
  38. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  39. 1 2 1 2 3 4 5 6 1 2 3
  40. 3 4 5 6 7 8 9 7 8 9 10 11 12 13 4 5 6 7 8 9 10
  41. 10 11 12 13 14 15 16 14 15 16 17 18 19 20 11 12 13 14 15 16 17
  42. 17 18 19 20 21 22 23 21 22 23 24 25 26 27 18 19 20 21 22 23 24
  43. 24 25 26 27 28 29 30 28 29 30 31 25 26 27 28 29 30
  44. 31
  45. October November December
  46. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  47. 1 1 2 3 4 5 1 2 3
  48. 2 3 4 5 6 7 8 6 7 8 9 10 11 12 4 5 6 7 8 9 10
  49. 9 10 11 12 13 14 15 13 14 15 16 17 18 19 11 12 13 14 15 16 17
  50. 16 17 18 19 20 21 22 20 21 22 23 24 25 26 18 19 20 21 22 23 24
  51. 23 24 25 26 27 28 29 27 28 29 30 25 26 27 28 29 30 31
  52. 30 31
  53. """
  54. result_2004_01_text = """\
  55. January 2004
  56. Mo Tu We Th Fr Sa Su
  57. 1 2 3 4
  58. 5 6 7 8 9 10 11
  59. 12 13 14 15 16 17 18
  60. 19 20 21 22 23 24 25
  61. 26 27 28 29 30 31
  62. """
  63. result_2004_text = """\
  64. 2004
  65. January February March
  66. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  67. 1 2 3 4 1 1 2 3 4 5 6 7
  68. 5 6 7 8 9 10 11 2 3 4 5 6 7 8 8 9 10 11 12 13 14
  69. 12 13 14 15 16 17 18 9 10 11 12 13 14 15 15 16 17 18 19 20 21
  70. 19 20 21 22 23 24 25 16 17 18 19 20 21 22 22 23 24 25 26 27 28
  71. 26 27 28 29 30 31 23 24 25 26 27 28 29 29 30 31
  72. April May June
  73. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  74. 1 2 3 4 1 2 1 2 3 4 5 6
  75. 5 6 7 8 9 10 11 3 4 5 6 7 8 9 7 8 9 10 11 12 13
  76. 12 13 14 15 16 17 18 10 11 12 13 14 15 16 14 15 16 17 18 19 20
  77. 19 20 21 22 23 24 25 17 18 19 20 21 22 23 21 22 23 24 25 26 27
  78. 26 27 28 29 30 24 25 26 27 28 29 30 28 29 30
  79. 31
  80. July August September
  81. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  82. 1 2 3 4 1 1 2 3 4 5
  83. 5 6 7 8 9 10 11 2 3 4 5 6 7 8 6 7 8 9 10 11 12
  84. 12 13 14 15 16 17 18 9 10 11 12 13 14 15 13 14 15 16 17 18 19
  85. 19 20 21 22 23 24 25 16 17 18 19 20 21 22 20 21 22 23 24 25 26
  86. 26 27 28 29 30 31 23 24 25 26 27 28 29 27 28 29 30
  87. 30 31
  88. October November December
  89. Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su
  90. 1 2 3 1 2 3 4 5 6 7 1 2 3 4 5
  91. 4 5 6 7 8 9 10 8 9 10 11 12 13 14 6 7 8 9 10 11 12
  92. 11 12 13 14 15 16 17 15 16 17 18 19 20 21 13 14 15 16 17 18 19
  93. 18 19 20 21 22 23 24 22 23 24 25 26 27 28 20 21 22 23 24 25 26
  94. 25 26 27 28 29 30 31 29 30 27 28 29 30 31
  95. """
  96. default_format = dict(year="year", month="month", encoding="ascii")
  97. result_2004_html = """\
  98. <?xml version="1.0" encoding="{encoding}"?>
  99. <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
  100. <html>
  101. <head>
  102. <meta http-equiv="Content-Type" content="text/html; charset={encoding}" />
  103. <link rel="stylesheet" type="text/css" href="calendar.css" />
  104. <title>Calendar for 2004</title>
  105. </head>
  106. <body>
  107. <table border="0" cellpadding="0" cellspacing="0" class="{year}">
  108. <tr><th colspan="3" class="{year}">2004</th></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  109. <tr><th colspan="7" class="{month}">January</th></tr>
  110. <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>
  111. <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>
  112. <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>
  113. <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>
  114. <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>
  115. <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>
  116. </table>
  117. </td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  118. <tr><th colspan="7" class="{month}">February</th></tr>
  119. <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>
  120. <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>
  121. <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>
  122. <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>
  123. <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>
  124. <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>
  125. </table>
  126. </td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  127. <tr><th colspan="7" class="{month}">March</th></tr>
  128. <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>
  129. <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>
  130. <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>
  131. <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>
  132. <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>
  133. <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>
  134. </table>
  135. </td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  136. <tr><th colspan="7" class="{month}">April</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="noday">&nbsp;</td><td class="thu">1</td><td class="fri">2</td><td class="sat">3</td><td class="sun">4</td></tr>
  139. <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>
  140. <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>
  141. <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>
  142. <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>
  143. </table>
  144. </td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  145. <tr><th colspan="7" class="{month}">May</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="noday">&nbsp;</td><td class="sat">1</td><td class="sun">2</td></tr>
  148. <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>
  149. <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>
  150. <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>
  151. <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>
  152. <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>
  153. </table>
  154. </td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  155. <tr><th colspan="7" class="{month}">June</th></tr>
  156. <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>
  157. <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>
  158. <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>
  159. <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>
  160. <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>
  161. <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>
  162. </table>
  163. </td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  164. <tr><th colspan="7" class="{month}">July</th></tr>
  165. <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>
  166. <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>
  167. <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>
  168. <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>
  169. <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>
  170. <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>
  171. </table>
  172. </td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  173. <tr><th colspan="7" class="{month}">August</th></tr>
  174. <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>
  175. <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>
  176. <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>
  177. <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>
  178. <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>
  179. <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>
  180. <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>
  181. </table>
  182. </td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  183. <tr><th colspan="7" class="{month}">September</th></tr>
  184. <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>
  185. <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>
  186. <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>
  187. <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>
  188. <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>
  189. <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>
  190. </table>
  191. </td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  192. <tr><th colspan="7" class="{month}">October</th></tr>
  193. <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>
  194. <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>
  195. <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>
  196. <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>
  197. <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>
  198. <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>
  199. </table>
  200. </td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  201. <tr><th colspan="7" class="{month}">November</th></tr>
  202. <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>
  203. <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>
  204. <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>
  205. <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>
  206. <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>
  207. <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>
  208. </table>
  209. </td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">
  210. <tr><th colspan="7" class="{month}">December</th></tr>
  211. <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>
  212. <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>
  213. <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>
  214. <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>
  215. <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>
  216. <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>
  217. </table>
  218. </td></tr></table></body>
  219. </html>
  220. """
  221. result_2004_days = [
  222. [[[0, 0, 0, 1, 2, 3, 4],
  223. [5, 6, 7, 8, 9, 10, 11],
  224. [12, 13, 14, 15, 16, 17, 18],
  225. [19, 20, 21, 22, 23, 24, 25],
  226. [26, 27, 28, 29, 30, 31, 0]],
  227. [[0, 0, 0, 0, 0, 0, 1],
  228. [2, 3, 4, 5, 6, 7, 8],
  229. [9, 10, 11, 12, 13, 14, 15],
  230. [16, 17, 18, 19, 20, 21, 22],
  231. [23, 24, 25, 26, 27, 28, 29]],
  232. [[1, 2, 3, 4, 5, 6, 7],
  233. [8, 9, 10, 11, 12, 13, 14],
  234. [15, 16, 17, 18, 19, 20, 21],
  235. [22, 23, 24, 25, 26, 27, 28],
  236. [29, 30, 31, 0, 0, 0, 0]]],
  237. [[[0, 0, 0, 1, 2, 3, 4],
  238. [5, 6, 7, 8, 9, 10, 11],
  239. [12, 13, 14, 15, 16, 17, 18],
  240. [19, 20, 21, 22, 23, 24, 25],
  241. [26, 27, 28, 29, 30, 0, 0]],
  242. [[0, 0, 0, 0, 0, 1, 2],
  243. [3, 4, 5, 6, 7, 8, 9],
  244. [10, 11, 12, 13, 14, 15, 16],
  245. [17, 18, 19, 20, 21, 22, 23],
  246. [24, 25, 26, 27, 28, 29, 30],
  247. [31, 0, 0, 0, 0, 0, 0]],
  248. [[0, 1, 2, 3, 4, 5, 6],
  249. [7, 8, 9, 10, 11, 12, 13],
  250. [14, 15, 16, 17, 18, 19, 20],
  251. [21, 22, 23, 24, 25, 26, 27],
  252. [28, 29, 30, 0, 0, 0, 0]]],
  253. [[[0, 0, 0, 1, 2, 3, 4],
  254. [5, 6, 7, 8, 9, 10, 11],
  255. [12, 13, 14, 15, 16, 17, 18],
  256. [19, 20, 21, 22, 23, 24, 25],
  257. [26, 27, 28, 29, 30, 31, 0]],
  258. [[0, 0, 0, 0, 0, 0, 1],
  259. [2, 3, 4, 5, 6, 7, 8],
  260. [9, 10, 11, 12, 13, 14, 15],
  261. [16, 17, 18, 19, 20, 21, 22],
  262. [23, 24, 25, 26, 27, 28, 29],
  263. [30, 31, 0, 0, 0, 0, 0]],
  264. [[0, 0, 1, 2, 3, 4, 5],
  265. [6, 7, 8, 9, 10, 11, 12],
  266. [13, 14, 15, 16, 17, 18, 19],
  267. [20, 21, 22, 23, 24, 25, 26],
  268. [27, 28, 29, 30, 0, 0, 0]]],
  269. [[[0, 0, 0, 0, 1, 2, 3],
  270. [4, 5, 6, 7, 8, 9, 10],
  271. [11, 12, 13, 14, 15, 16, 17],
  272. [18, 19, 20, 21, 22, 23, 24],
  273. [25, 26, 27, 28, 29, 30, 31]],
  274. [[1, 2, 3, 4, 5, 6, 7],
  275. [8, 9, 10, 11, 12, 13, 14],
  276. [15, 16, 17, 18, 19, 20, 21],
  277. [22, 23, 24, 25, 26, 27, 28],
  278. [29, 30, 0, 0, 0, 0, 0]],
  279. [[0, 0, 1, 2, 3, 4, 5],
  280. [6, 7, 8, 9, 10, 11, 12],
  281. [13, 14, 15, 16, 17, 18, 19],
  282. [20, 21, 22, 23, 24, 25, 26],
  283. [27, 28, 29, 30, 31, 0, 0]]]
  284. ]
  285. result_2004_dates = \
  286. [[['12/29/03 12/30/03 12/31/03 01/01/04 01/02/04 01/03/04 01/04/04',
  287. '01/05/04 01/06/04 01/07/04 01/08/04 01/09/04 01/10/04 01/11/04',
  288. '01/12/04 01/13/04 01/14/04 01/15/04 01/16/04 01/17/04 01/18/04',
  289. '01/19/04 01/20/04 01/21/04 01/22/04 01/23/04 01/24/04 01/25/04',
  290. '01/26/04 01/27/04 01/28/04 01/29/04 01/30/04 01/31/04 02/01/04'],
  291. ['01/26/04 01/27/04 01/28/04 01/29/04 01/30/04 01/31/04 02/01/04',
  292. '02/02/04 02/03/04 02/04/04 02/05/04 02/06/04 02/07/04 02/08/04',
  293. '02/09/04 02/10/04 02/11/04 02/12/04 02/13/04 02/14/04 02/15/04',
  294. '02/16/04 02/17/04 02/18/04 02/19/04 02/20/04 02/21/04 02/22/04',
  295. '02/23/04 02/24/04 02/25/04 02/26/04 02/27/04 02/28/04 02/29/04'],
  296. ['03/01/04 03/02/04 03/03/04 03/04/04 03/05/04 03/06/04 03/07/04',
  297. '03/08/04 03/09/04 03/10/04 03/11/04 03/12/04 03/13/04 03/14/04',
  298. '03/15/04 03/16/04 03/17/04 03/18/04 03/19/04 03/20/04 03/21/04',
  299. '03/22/04 03/23/04 03/24/04 03/25/04 03/26/04 03/27/04 03/28/04',
  300. '03/29/04 03/30/04 03/31/04 04/01/04 04/02/04 04/03/04 04/04/04']],
  301. [['03/29/04 03/30/04 03/31/04 04/01/04 04/02/04 04/03/04 04/04/04',
  302. '04/05/04 04/06/04 04/07/04 04/08/04 04/09/04 04/10/04 04/11/04',
  303. '04/12/04 04/13/04 04/14/04 04/15/04 04/16/04 04/17/04 04/18/04',
  304. '04/19/04 04/20/04 04/21/04 04/22/04 04/23/04 04/24/04 04/25/04',
  305. '04/26/04 04/27/04 04/28/04 04/29/04 04/30/04 05/01/04 05/02/04'],
  306. ['04/26/04 04/27/04 04/28/04 04/29/04 04/30/04 05/01/04 05/02/04',
  307. '05/03/04 05/04/04 05/05/04 05/06/04 05/07/04 05/08/04 05/09/04',
  308. '05/10/04 05/11/04 05/12/04 05/13/04 05/14/04 05/15/04 05/16/04',
  309. '05/17/04 05/18/04 05/19/04 05/20/04 05/21/04 05/22/04 05/23/04',
  310. '05/24/04 05/25/04 05/26/04 05/27/04 05/28/04 05/29/04 05/30/04',
  311. '05/31/04 06/01/04 06/02/04 06/03/04 06/04/04 06/05/04 06/06/04'],
  312. ['05/31/04 06/01/04 06/02/04 06/03/04 06/04/04 06/05/04 06/06/04',
  313. '06/07/04 06/08/04 06/09/04 06/10/04 06/11/04 06/12/04 06/13/04',
  314. '06/14/04 06/15/04 06/16/04 06/17/04 06/18/04 06/19/04 06/20/04',
  315. '06/21/04 06/22/04 06/23/04 06/24/04 06/25/04 06/26/04 06/27/04',
  316. '06/28/04 06/29/04 06/30/04 07/01/04 07/02/04 07/03/04 07/04/04']],
  317. [['06/28/04 06/29/04 06/30/04 07/01/04 07/02/04 07/03/04 07/04/04',
  318. '07/05/04 07/06/04 07/07/04 07/08/04 07/09/04 07/10/04 07/11/04',
  319. '07/12/04 07/13/04 07/14/04 07/15/04 07/16/04 07/17/04 07/18/04',
  320. '07/19/04 07/20/04 07/21/04 07/22/04 07/23/04 07/24/04 07/25/04',
  321. '07/26/04 07/27/04 07/28/04 07/29/04 07/30/04 07/31/04 08/01/04'],
  322. ['07/26/04 07/27/04 07/28/04 07/29/04 07/30/04 07/31/04 08/01/04',
  323. '08/02/04 08/03/04 08/04/04 08/05/04 08/06/04 08/07/04 08/08/04',
  324. '08/09/04 08/10/04 08/11/04 08/12/04 08/13/04 08/14/04 08/15/04',
  325. '08/16/04 08/17/04 08/18/04 08/19/04 08/20/04 08/21/04 08/22/04',
  326. '08/23/04 08/24/04 08/25/04 08/26/04 08/27/04 08/28/04 08/29/04',
  327. '08/30/04 08/31/04 09/01/04 09/02/04 09/03/04 09/04/04 09/05/04'],
  328. ['08/30/04 08/31/04 09/01/04 09/02/04 09/03/04 09/04/04 09/05/04',
  329. '09/06/04 09/07/04 09/08/04 09/09/04 09/10/04 09/11/04 09/12/04',
  330. '09/13/04 09/14/04 09/15/04 09/16/04 09/17/04 09/18/04 09/19/04',
  331. '09/20/04 09/21/04 09/22/04 09/23/04 09/24/04 09/25/04 09/26/04',
  332. '09/27/04 09/28/04 09/29/04 09/30/04 10/01/04 10/02/04 10/03/04']],
  333. [['09/27/04 09/28/04 09/29/04 09/30/04 10/01/04 10/02/04 10/03/04',
  334. '10/04/04 10/05/04 10/06/04 10/07/04 10/08/04 10/09/04 10/10/04',
  335. '10/11/04 10/12/04 10/13/04 10/14/04 10/15/04 10/16/04 10/17/04',
  336. '10/18/04 10/19/04 10/20/04 10/21/04 10/22/04 10/23/04 10/24/04',
  337. '10/25/04 10/26/04 10/27/04 10/28/04 10/29/04 10/30/04 10/31/04'],
  338. ['11/01/04 11/02/04 11/03/04 11/04/04 11/05/04 11/06/04 11/07/04',
  339. '11/08/04 11/09/04 11/10/04 11/11/04 11/12/04 11/13/04 11/14/04',
  340. '11/15/04 11/16/04 11/17/04 11/18/04 11/19/04 11/20/04 11/21/04',
  341. '11/22/04 11/23/04 11/24/04 11/25/04 11/26/04 11/27/04 11/28/04',
  342. '11/29/04 11/30/04 12/01/04 12/02/04 12/03/04 12/04/04 12/05/04'],
  343. ['11/29/04 11/30/04 12/01/04 12/02/04 12/03/04 12/04/04 12/05/04',
  344. '12/06/04 12/07/04 12/08/04 12/09/04 12/10/04 12/11/04 12/12/04',
  345. '12/13/04 12/14/04 12/15/04 12/16/04 12/17/04 12/18/04 12/19/04',
  346. '12/20/04 12/21/04 12/22/04 12/23/04 12/24/04 12/25/04 12/26/04',
  347. '12/27/04 12/28/04 12/29/04 12/30/04 12/31/04 01/01/05 01/02/05']]]
  348. class OutputTestCase(unittest.TestCase):
  349. def normalize_calendar(self, s):
  350. # Filters out locale dependent strings
  351. def neitherspacenordigit(c):
  352. return not c.isspace() and not c.isdigit()
  353. lines = []
  354. for line in s.splitlines(keepends=False):
  355. # Drop texts, as they are locale dependent
  356. if line and not filter(neitherspacenordigit, line):
  357. lines.append(line)
  358. return lines
  359. def check_htmlcalendar_encoding(self, req, res):
  360. cal = calendar.HTMLCalendar()
  361. format_ = default_format.copy()
  362. format_["encoding"] = req or 'utf-8'
  363. output = cal.formatyearpage(2004, encoding=req)
  364. self.assertEqual(
  365. output,
  366. result_2004_html.format(**format_).encode(res)
  367. )
  368. def test_output(self):
  369. self.assertEqual(
  370. self.normalize_calendar(calendar.calendar(2004)),
  371. self.normalize_calendar(result_2004_text)
  372. )
  373. self.assertEqual(
  374. self.normalize_calendar(calendar.calendar(0)),
  375. self.normalize_calendar(result_0_text)
  376. )
  377. def test_output_textcalendar(self):
  378. self.assertEqual(
  379. calendar.TextCalendar().formatyear(2004),
  380. result_2004_text
  381. )
  382. self.assertEqual(
  383. calendar.TextCalendar().formatyear(0),
  384. result_0_text
  385. )
  386. def test_output_htmlcalendar_encoding_ascii(self):
  387. self.check_htmlcalendar_encoding('ascii', 'ascii')
  388. def test_output_htmlcalendar_encoding_utf8(self):
  389. self.check_htmlcalendar_encoding('utf-8', 'utf-8')
  390. def test_output_htmlcalendar_encoding_default(self):
  391. self.check_htmlcalendar_encoding(None, sys.getdefaultencoding())
  392. def test_yeardatescalendar(self):
  393. def shrink(cal):
  394. return [[[' '.join('{:02d}/{:02d}/{}'.format(
  395. d.month, d.day, str(d.year)[-2:]) for d in z)
  396. for z in y] for y in x] for x in cal]
  397. self.assertEqual(
  398. shrink(calendar.Calendar().yeardatescalendar(2004)),
  399. result_2004_dates
  400. )
  401. def test_yeardayscalendar(self):
  402. self.assertEqual(
  403. calendar.Calendar().yeardayscalendar(2004),
  404. result_2004_days
  405. )
  406. def test_formatweekheader_short(self):
  407. self.assertEqual(
  408. calendar.TextCalendar().formatweekheader(2),
  409. 'Mo Tu We Th Fr Sa Su'
  410. )
  411. def test_formatweekheader_long(self):
  412. self.assertEqual(
  413. calendar.TextCalendar().formatweekheader(9),
  414. ' Monday Tuesday Wednesday Thursday '
  415. ' Friday Saturday Sunday '
  416. )
  417. def test_formatmonth(self):
  418. self.assertEqual(
  419. calendar.TextCalendar().formatmonth(2004, 1),
  420. result_2004_01_text
  421. )
  422. self.assertEqual(
  423. calendar.TextCalendar().formatmonth(0, 2),
  424. result_0_02_text
  425. )
  426. def test_formatmonthname_with_year(self):
  427. self.assertEqual(
  428. calendar.HTMLCalendar().formatmonthname(2004, 1, withyear=True),
  429. '<tr><th colspan="7" class="month">January 2004</th></tr>'
  430. )
  431. def test_formatmonthname_without_year(self):
  432. self.assertEqual(
  433. calendar.HTMLCalendar().formatmonthname(2004, 1, withyear=False),
  434. '<tr><th colspan="7" class="month">January</th></tr>'
  435. )
  436. def test_prweek(self):
  437. with support.captured_stdout() as out:
  438. week = [(1,0), (2,1), (3,2), (4,3), (5,4), (6,5), (7,6)]
  439. calendar.TextCalendar().prweek(week, 1)
  440. self.assertEqual(out.getvalue(), " 1 2 3 4 5 6 7")
  441. def test_prmonth(self):
  442. with support.captured_stdout() as out:
  443. calendar.TextCalendar().prmonth(2004, 1)
  444. self.assertEqual(out.getvalue(), result_2004_01_text)
  445. def test_pryear(self):
  446. with support.captured_stdout() as out:
  447. calendar.TextCalendar().pryear(2004)
  448. self.assertEqual(out.getvalue(), result_2004_text)
  449. def test_format(self):
  450. with support.captured_stdout() as out:
  451. calendar.format(["1", "2", "3"], colwidth=3, spacing=1)
  452. self.assertEqual(out.getvalue().strip(), "1 2 3")
  453. class CalendarTestCase(unittest.TestCase):
  454. def test_isleap(self):
  455. # Make sure that the return is right for a few years, and
  456. # ensure that the return values are 1 or 0, not just true or
  457. # false (see SF bug #485794). Specific additional tests may
  458. # be appropriate; this tests a single "cycle".
  459. self.assertEqual(calendar.isleap(2000), 1)
  460. self.assertEqual(calendar.isleap(2001), 0)
  461. self.assertEqual(calendar.isleap(2002), 0)
  462. self.assertEqual(calendar.isleap(2003), 0)
  463. def test_setfirstweekday(self):
  464. self.assertRaises(TypeError, calendar.setfirstweekday, 'flabber')
  465. self.assertRaises(ValueError, calendar.setfirstweekday, -1)
  466. self.assertRaises(ValueError, calendar.setfirstweekday, 200)
  467. orig = calendar.firstweekday()
  468. calendar.setfirstweekday(calendar.SUNDAY)
  469. self.assertEqual(calendar.firstweekday(), calendar.SUNDAY)
  470. calendar.setfirstweekday(calendar.MONDAY)
  471. self.assertEqual(calendar.firstweekday(), calendar.MONDAY)
  472. calendar.setfirstweekday(orig)
  473. def test_illegal_weekday_reported(self):
  474. with self.assertRaisesRegex(calendar.IllegalWeekdayError, '123'):
  475. calendar.setfirstweekday(123)
  476. def test_enumerate_weekdays(self):
  477. self.assertRaises(IndexError, calendar.day_abbr.__getitem__, -10)
  478. self.assertRaises(IndexError, calendar.day_name.__getitem__, 10)
  479. self.assertEqual(len([d for d in calendar.day_abbr]), 7)
  480. def test_days(self):
  481. for attr in "day_name", "day_abbr":
  482. value = getattr(calendar, attr)
  483. self.assertEqual(len(value), 7)
  484. self.assertEqual(len(value[:]), 7)
  485. # ensure they're all unique
  486. self.assertEqual(len(set(value)), 7)
  487. # verify it "acts like a sequence" in two forms of iteration
  488. self.assertEqual(value[::-1], list(reversed(value)))
  489. def test_months(self):
  490. for attr in "month_name", "month_abbr":
  491. value = getattr(calendar, attr)
  492. self.assertEqual(len(value), 13)
  493. self.assertEqual(len(value[:]), 13)
  494. self.assertEqual(value[0], "")
  495. # ensure they're all unique
  496. self.assertEqual(len(set(value)), 13)
  497. # verify it "acts like a sequence" in two forms of iteration
  498. self.assertEqual(value[::-1], list(reversed(value)))
  499. def test_locale_calendars(self):
  500. # ensure that Locale{Text,HTML}Calendar resets the locale properly
  501. # (it is still not thread-safe though)
  502. old_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)
  503. try:
  504. cal = calendar.LocaleTextCalendar(locale='')
  505. local_weekday = cal.formatweekday(1, 10)
  506. local_month = cal.formatmonthname(2010, 10, 10)
  507. except locale.Error:
  508. # cannot set the system default locale -- skip rest of test
  509. raise unittest.SkipTest('cannot set the system default locale')
  510. self.assertIsInstance(local_weekday, str)
  511. self.assertIsInstance(local_month, str)
  512. self.assertEqual(len(local_weekday), 10)
  513. self.assertGreaterEqual(len(local_month), 10)
  514. cal = calendar.LocaleHTMLCalendar(locale='')
  515. local_weekday = cal.formatweekday(1)
  516. local_month = cal.formatmonthname(2010, 10)
  517. self.assertIsInstance(local_weekday, str)
  518. self.assertIsInstance(local_month, str)
  519. new_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)
  520. self.assertEqual(old_october, new_october)
  521. def test_locale_calendar_formatweekday(self):
  522. try:
  523. # formatweekday uses different day names based on the available width.
  524. cal = calendar.LocaleTextCalendar(locale='en_US')
  525. # For short widths, a centered, abbreviated name is used.
  526. self.assertEqual(cal.formatweekday(0, 5), " Mon ")
  527. # For really short widths, even the abbreviated name is truncated.
  528. self.assertEqual(cal.formatweekday(0, 2), "Mo")
  529. # For long widths, the full day name is used.
  530. self.assertEqual(cal.formatweekday(0, 10), " Monday ")
  531. except locale.Error:
  532. raise unittest.SkipTest('cannot set the en_US locale')
  533. def test_locale_html_calendar_custom_css_class_month_name(self):
  534. try:
  535. cal = calendar.LocaleHTMLCalendar(locale='')
  536. local_month = cal.formatmonthname(2010, 10, 10)
  537. except locale.Error:
  538. # cannot set the system default locale -- skip rest of test
  539. raise unittest.SkipTest('cannot set the system default locale')
  540. self.assertIn('class="month"', local_month)
  541. cal.cssclass_month_head = "text-center month"
  542. local_month = cal.formatmonthname(2010, 10, 10)
  543. self.assertIn('class="text-center month"', local_month)
  544. def test_locale_html_calendar_custom_css_class_weekday(self):
  545. try:
  546. cal = calendar.LocaleHTMLCalendar(locale='')
  547. local_weekday = cal.formatweekday(6)
  548. except locale.Error:
  549. # cannot set the system default locale -- skip rest of test
  550. raise unittest.SkipTest('cannot set the system default locale')
  551. self.assertIn('class="sun"', local_weekday)
  552. cal.cssclasses_weekday_head = ["mon2", "tue2", "wed2", "thu2", "fri2", "sat2", "sun2"]
  553. local_weekday = cal.formatweekday(6)
  554. self.assertIn('class="sun2"', local_weekday)
  555. def test_itermonthdays3(self):
  556. # ensure itermonthdays3 doesn't overflow after datetime.MAXYEAR
  557. list(calendar.Calendar().itermonthdays3(datetime.MAXYEAR, 12))
  558. def test_itermonthdays4(self):
  559. cal = calendar.Calendar(firstweekday=3)
  560. days = list(cal.itermonthdays4(2001, 2))
  561. self.assertEqual(days[0], (2001, 2, 1, 3))
  562. self.assertEqual(days[-1], (2001, 2, 28, 2))
  563. def test_itermonthdays(self):
  564. for firstweekday in range(7):
  565. cal = calendar.Calendar(firstweekday)
  566. # Test the extremes, see #28253 and #26650
  567. for y, m in [(1, 1), (9999, 12)]:
  568. days = list(cal.itermonthdays(y, m))
  569. self.assertIn(len(days), (35, 42))
  570. # Test a short month
  571. cal = calendar.Calendar(firstweekday=3)
  572. days = list(cal.itermonthdays(2001, 2))
  573. self.assertEqual(days, list(range(1, 29)))
  574. def test_itermonthdays2(self):
  575. for firstweekday in range(7):
  576. cal = calendar.Calendar(firstweekday)
  577. # Test the extremes, see #28253 and #26650
  578. for y, m in [(1, 1), (9999, 12)]:
  579. days = list(cal.itermonthdays2(y, m))
  580. self.assertEqual(days[0][1], firstweekday)
  581. self.assertEqual(days[-1][1], (firstweekday - 1) % 7)
  582. def test_iterweekdays(self):
  583. week0 = list(range(7))
  584. for firstweekday in range(7):
  585. cal = calendar.Calendar(firstweekday)
  586. week = list(cal.iterweekdays())
  587. expected = week0[firstweekday:] + week0[:firstweekday]
  588. self.assertEqual(week, expected)
  589. class MonthCalendarTestCase(unittest.TestCase):
  590. def setUp(self):
  591. self.oldfirstweekday = calendar.firstweekday()
  592. calendar.setfirstweekday(self.firstweekday)
  593. def tearDown(self):
  594. calendar.setfirstweekday(self.oldfirstweekday)
  595. def check_weeks(self, year, month, weeks):
  596. cal = calendar.monthcalendar(year, month)
  597. self.assertEqual(len(cal), len(weeks))
  598. for i in range(len(weeks)):
  599. self.assertEqual(weeks[i], sum(day != 0 for day in cal[i]))
  600. class MondayTestCase(MonthCalendarTestCase):
  601. firstweekday = calendar.MONDAY
  602. def test_february(self):
  603. # A 28-day february starting on monday (7+7+7+7 days)
  604. self.check_weeks(1999, 2, (7, 7, 7, 7))
  605. # A 28-day february starting on tuesday (6+7+7+7+1 days)
  606. self.check_weeks(2005, 2, (6, 7, 7, 7, 1))
  607. # A 28-day february starting on sunday (1+7+7+7+6 days)
  608. self.check_weeks(1987, 2, (1, 7, 7, 7, 6))
  609. # A 29-day february starting on monday (7+7+7+7+1 days)
  610. self.check_weeks(1988, 2, (7, 7, 7, 7, 1))
  611. # A 29-day february starting on tuesday (6+7+7+7+2 days)
  612. self.check_weeks(1972, 2, (6, 7, 7, 7, 2))
  613. # A 29-day february starting on sunday (1+7+7+7+7 days)
  614. self.check_weeks(2004, 2, (1, 7, 7, 7, 7))
  615. def test_april(self):
  616. # A 30-day april starting on monday (7+7+7+7+2 days)
  617. self.check_weeks(1935, 4, (7, 7, 7, 7, 2))
  618. # A 30-day april starting on tuesday (6+7+7+7+3 days)
  619. self.check_weeks(1975, 4, (6, 7, 7, 7, 3))
  620. # A 30-day april starting on sunday (1+7+7+7+7+1 days)
  621. self.check_weeks(1945, 4, (1, 7, 7, 7, 7, 1))
  622. # A 30-day april starting on saturday (2+7+7+7+7 days)
  623. self.check_weeks(1995, 4, (2, 7, 7, 7, 7))
  624. # A 30-day april starting on friday (3+7+7+7+6 days)
  625. self.check_weeks(1994, 4, (3, 7, 7, 7, 6))
  626. def test_december(self):
  627. # A 31-day december starting on monday (7+7+7+7+3 days)
  628. self.check_weeks(1980, 12, (7, 7, 7, 7, 3))
  629. # A 31-day december starting on tuesday (6+7+7+7+4 days)
  630. self.check_weeks(1987, 12, (6, 7, 7, 7, 4))
  631. # A 31-day december starting on sunday (1+7+7+7+7+2 days)
  632. self.check_weeks(1968, 12, (1, 7, 7, 7, 7, 2))
  633. # A 31-day december starting on thursday (4+7+7+7+6 days)
  634. self.check_weeks(1988, 12, (4, 7, 7, 7, 6))
  635. # A 31-day december starting on friday (3+7+7+7+7 days)
  636. self.check_weeks(2017, 12, (3, 7, 7, 7, 7))
  637. # A 31-day december starting on saturday (2+7+7+7+7+1 days)
  638. self.check_weeks(2068, 12, (2, 7, 7, 7, 7, 1))
  639. class SundayTestCase(MonthCalendarTestCase):
  640. firstweekday = calendar.SUNDAY
  641. def test_february(self):
  642. # A 28-day february starting on sunday (7+7+7+7 days)
  643. self.check_weeks(2009, 2, (7, 7, 7, 7))
  644. # A 28-day february starting on monday (6+7+7+7+1 days)
  645. self.check_weeks(1999, 2, (6, 7, 7, 7, 1))
  646. # A 28-day february starting on saturday (1+7+7+7+6 days)
  647. self.check_weeks(1997, 2, (1, 7, 7, 7, 6))
  648. # A 29-day february starting on sunday (7+7+7+7+1 days)
  649. self.check_weeks(2004, 2, (7, 7, 7, 7, 1))
  650. # A 29-day february starting on monday (6+7+7+7+2 days)
  651. self.check_weeks(1960, 2, (6, 7, 7, 7, 2))
  652. # A 29-day february starting on saturday (1+7+7+7+7 days)
  653. self.check_weeks(1964, 2, (1, 7, 7, 7, 7))
  654. def test_april(self):
  655. # A 30-day april starting on sunday (7+7+7+7+2 days)
  656. self.check_weeks(1923, 4, (7, 7, 7, 7, 2))
  657. # A 30-day april starting on monday (6+7+7+7+3 days)
  658. self.check_weeks(1918, 4, (6, 7, 7, 7, 3))
  659. # A 30-day april starting on saturday (1+7+7+7+7+1 days)
  660. self.check_weeks(1950, 4, (1, 7, 7, 7, 7, 1))
  661. # A 30-day april starting on friday (2+7+7+7+7 days)
  662. self.check_weeks(1960, 4, (2, 7, 7, 7, 7))
  663. # A 30-day april starting on thursday (3+7+7+7+6 days)
  664. self.check_weeks(1909, 4, (3, 7, 7, 7, 6))
  665. def test_december(self):
  666. # A 31-day december starting on sunday (7+7+7+7+3 days)
  667. self.check_weeks(2080, 12, (7, 7, 7, 7, 3))
  668. # A 31-day december starting on monday (6+7+7+7+4 days)
  669. self.check_weeks(1941, 12, (6, 7, 7, 7, 4))
  670. # A 31-day december starting on saturday (1+7+7+7+7+2 days)
  671. self.check_weeks(1923, 12, (1, 7, 7, 7, 7, 2))
  672. # A 31-day december starting on wednesday (4+7+7+7+6 days)
  673. self.check_weeks(1948, 12, (4, 7, 7, 7, 6))
  674. # A 31-day december starting on thursday (3+7+7+7+7 days)
  675. self.check_weeks(1927, 12, (3, 7, 7, 7, 7))
  676. # A 31-day december starting on friday (2+7+7+7+7+1 days)
  677. self.check_weeks(1995, 12, (2, 7, 7, 7, 7, 1))
  678. class TimegmTestCase(unittest.TestCase):
  679. TIMESTAMPS = [0, 10, 100, 1000, 10000, 100000, 1000000,
  680. 1234567890, 1262304000, 1275785153,]
  681. def test_timegm(self):
  682. for secs in self.TIMESTAMPS:
  683. tuple = time.gmtime(secs)
  684. self.assertEqual(secs, calendar.timegm(tuple))
  685. class MonthRangeTestCase(unittest.TestCase):
  686. def test_january(self):
  687. # Tests valid lower boundary case.
  688. self.assertEqual(calendar.monthrange(2004,1), (3,31))
  689. def test_february_leap(self):
  690. # Tests February during leap year.
  691. self.assertEqual(calendar.monthrange(2004,2), (6,29))
  692. def test_february_nonleap(self):
  693. # Tests February in non-leap year.
  694. self.assertEqual(calendar.monthrange(2010,2), (0,28))
  695. def test_december(self):
  696. # Tests valid upper boundary case.
  697. self.assertEqual(calendar.monthrange(2004,12), (2,31))
  698. def test_zeroth_month(self):
  699. # Tests low invalid boundary case.
  700. with self.assertRaises(calendar.IllegalMonthError):
  701. calendar.monthrange(2004, 0)
  702. def test_thirteenth_month(self):
  703. # Tests high invalid boundary case.
  704. with self.assertRaises(calendar.IllegalMonthError):
  705. calendar.monthrange(2004, 13)
  706. def test_illegal_month_reported(self):
  707. with self.assertRaisesRegex(calendar.IllegalMonthError, '65'):
  708. calendar.monthrange(2004, 65)
  709. class LeapdaysTestCase(unittest.TestCase):
  710. def test_no_range(self):
  711. # test when no range i.e. two identical years as args
  712. self.assertEqual(calendar.leapdays(2010,2010), 0)
  713. def test_no_leapdays(self):
  714. # test when no leap years in range
  715. self.assertEqual(calendar.leapdays(2010,2011), 0)
  716. def test_no_leapdays_upper_boundary(self):
  717. # test no leap years in range, when upper boundary is a leap year
  718. self.assertEqual(calendar.leapdays(2010,2012), 0)
  719. def test_one_leapday_lower_boundary(self):
  720. # test when one leap year in range, lower boundary is leap year
  721. self.assertEqual(calendar.leapdays(2012,2013), 1)
  722. def test_several_leapyears_in_range(self):
  723. self.assertEqual(calendar.leapdays(1997,2020), 5)
  724. def conv(s):
  725. return s.replace('\n', os.linesep).encode()
  726. class CommandLineTestCase(unittest.TestCase):
  727. def run_ok(self, *args):
  728. return assert_python_ok('-m', 'calendar', *args)[1]
  729. def assertFailure(self, *args):
  730. rc, stdout, stderr = assert_python_failure('-m', 'calendar', *args)
  731. self.assertIn(b'usage:', stderr)
  732. self.assertEqual(rc, 2)
  733. def test_help(self):
  734. stdout = self.run_ok('-h')
  735. self.assertIn(b'usage:', stdout)
  736. self.assertIn(b'calendar.py', stdout)
  737. self.assertIn(b'--help', stdout)
  738. def test_illegal_arguments(self):
  739. self.assertFailure('-z')
  740. self.assertFailure('spam')
  741. self.assertFailure('2004', 'spam')
  742. self.assertFailure('-t', 'html', '2004', '1')
  743. def test_output_current_year(self):
  744. stdout = self.run_ok()
  745. year = datetime.datetime.now().year
  746. self.assertIn((' %s' % year).encode(), stdout)
  747. self.assertIn(b'January', stdout)
  748. self.assertIn(b'Mo Tu We Th Fr Sa Su', stdout)
  749. def test_output_year(self):
  750. stdout = self.run_ok('2004')
  751. self.assertEqual(stdout, conv(result_2004_text))
  752. def test_output_month(self):
  753. stdout = self.run_ok('2004', '1')
  754. self.assertEqual(stdout, conv(result_2004_01_text))
  755. def test_option_encoding(self):
  756. self.assertFailure('-e')
  757. self.assertFailure('--encoding')
  758. stdout = self.run_ok('--encoding', 'utf-16-le', '2004')
  759. self.assertEqual(stdout, result_2004_text.encode('utf-16-le'))
  760. def test_option_locale(self):
  761. self.assertFailure('-L')
  762. self.assertFailure('--locale')
  763. self.assertFailure('-L', 'en')
  764. lang, enc = locale.getlocale()
  765. lang = lang or 'C'
  766. enc = enc or 'UTF-8'
  767. try:
  768. oldlocale = locale.getlocale(locale.LC_TIME)
  769. try:
  770. locale.setlocale(locale.LC_TIME, (lang, enc))
  771. finally:
  772. locale.setlocale(locale.LC_TIME, oldlocale)
  773. except (locale.Error, ValueError):
  774. self.skipTest('cannot set the system default locale')
  775. stdout = self.run_ok('--locale', lang, '--encoding', enc, '2004')
  776. self.assertIn('2004'.encode(enc), stdout)
  777. def test_option_width(self):
  778. self.assertFailure('-w')
  779. self.assertFailure('--width')
  780. self.assertFailure('-w', 'spam')
  781. stdout = self.run_ok('--width', '3', '2004')
  782. self.assertIn(b'Mon Tue Wed Thu Fri Sat Sun', stdout)
  783. def test_option_lines(self):
  784. self.assertFailure('-l')
  785. self.assertFailure('--lines')
  786. self.assertFailure('-l', 'spam')
  787. stdout = self.run_ok('--lines', '2', '2004')
  788. self.assertIn(conv('December\n\nMo Tu We'), stdout)
  789. def test_option_spacing(self):
  790. self.assertFailure('-s')
  791. self.assertFailure('--spacing')
  792. self.assertFailure('-s', 'spam')
  793. stdout = self.run_ok('--spacing', '8', '2004')
  794. self.assertIn(b'Su Mo', stdout)
  795. def test_option_months(self):
  796. self.assertFailure('-m')
  797. self.assertFailure('--month')
  798. self.assertFailure('-m', 'spam')
  799. stdout = self.run_ok('--months', '1', '2004')
  800. self.assertIn(conv('\nMo Tu We Th Fr Sa Su\n'), stdout)
  801. def test_option_type(self):
  802. self.assertFailure('-t')
  803. self.assertFailure('--type')
  804. self.assertFailure('-t', 'spam')
  805. stdout = self.run_ok('--type', 'text', '2004')
  806. self.assertEqual(stdout, conv(result_2004_text))
  807. stdout = self.run_ok('--type', 'html', '2004')
  808. self.assertEqual(stdout[:6], b'<?xml ')
  809. self.assertIn(b'<title>Calendar for 2004</title>', stdout)
  810. def test_html_output_current_year(self):
  811. stdout = self.run_ok('--type', 'html')
  812. year = datetime.datetime.now().year
  813. self.assertIn(('<title>Calendar for %s</title>' % year).encode(),
  814. stdout)
  815. self.assertIn(b'<tr><th colspan="7" class="month">January</th></tr>',
  816. stdout)
  817. def test_html_output_year_encoding(self):
  818. stdout = self.run_ok('-t', 'html', '--encoding', 'ascii', '2004')
  819. self.assertEqual(stdout,
  820. result_2004_html.format(**default_format).encode('ascii'))
  821. def test_html_output_year_css(self):
  822. self.assertFailure('-t', 'html', '-c')
  823. self.assertFailure('-t', 'html', '--css')
  824. stdout = self.run_ok('-t', 'html', '--css', 'custom.css', '2004')
  825. self.assertIn(b'<link rel="stylesheet" type="text/css" '
  826. b'href="custom.css" />', stdout)
  827. class MiscTestCase(unittest.TestCase):
  828. def test__all__(self):
  829. not_exported = {
  830. 'mdays', 'January', 'February', 'EPOCH',
  831. 'different_locale', 'c', 'prweek', 'week', 'format',
  832. 'formatstring', 'main', 'monthlen', 'prevmonth', 'nextmonth'}
  833. support.check__all__(self, calendar, not_exported=not_exported)
  834. class TestSubClassingCase(unittest.TestCase):
  835. def setUp(self):
  836. class CustomHTMLCal(calendar.HTMLCalendar):
  837. cssclasses = [style + " text-nowrap" for style in
  838. calendar.HTMLCalendar.cssclasses]
  839. cssclasses_weekday_head = ["red", "blue", "green", "lilac",
  840. "yellow", "orange", "pink"]
  841. cssclass_month_head = "text-center month-head"
  842. cssclass_month = "text-center month"
  843. cssclass_year = "text-italic "
  844. cssclass_year_head = "lead "
  845. self.cal = CustomHTMLCal()
  846. def test_formatmonthname(self):
  847. self.assertIn('class="text-center month-head"',
  848. self.cal.formatmonthname(2017, 5))
  849. def test_formatmonth(self):
  850. self.assertIn('class="text-center month"',
  851. self.cal.formatmonth(2017, 5))
  852. def test_formatweek(self):
  853. weeks = self.cal.monthdays2calendar(2017, 5)
  854. self.assertIn('class="wed text-nowrap"', self.cal.formatweek(weeks[0]))
  855. def test_formatweek_head(self):
  856. header = self.cal.formatweekheader()
  857. for color in self.cal.cssclasses_weekday_head:
  858. self.assertIn('<th class="%s">' % color, header)
  859. def test_format_year(self):
  860. self.assertIn(
  861. ('<table border="0" cellpadding="0" cellspacing="0" class="%s">' %
  862. self.cal.cssclass_year), self.cal.formatyear(2017))
  863. def test_format_year_head(self):
  864. self.assertIn('<tr><th colspan="%d" class="%s">%s</th></tr>' % (
  865. 3, self.cal.cssclass_year_head, 2017), self.cal.formatyear(2017))
  866. if __name__ == "__main__":
  867. unittest.main()