mirror of
https://github.com/python/cpython.git
synced 2025-07-24 03:35:53 +00:00
8b7a4cc40e
Several class attributes have been added to calendar.HTMLCalendar that allow customization of the CSS classes used in the resulting HTML. This can be done by subclasses HTMLCalendar and overwriting those class attributes (Patch by Oz Tiram).
896 lines
46 KiB
Python
896 lines
46 KiB
Python
import calendar
|
|
import unittest
|
|
|
|
from test import support
|
|
from test.support.script_helper import assert_python_ok, assert_python_failure
|
|
import time
|
|
import locale
|
|
import sys
|
|
import datetime
|
|
import os
|
|
|
|
result_2004_01_text = """\
|
|
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
|
|
"""
|
|
|
|
result_2004_text = """\
|
|
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
|
|
"""
|
|
|
|
|
|
default_format = dict(year="year", month="month", encoding="ascii")
|
|
|
|
result_2004_html = """\
|
|
<?xml version="1.0" encoding="{encoding}"?>
|
|
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
|
|
<html>
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/html; charset={encoding}" />
|
|
<link rel="stylesheet" type="text/css" href="calendar.css" />
|
|
<title>Calendar for 2004</title>
|
|
</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"> </td><td class="noday"> </td><td class="noday"> </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"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </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"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </td><td class="noday"> </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"> </td><td class="noday"> </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"> </td><td class="noday"> </td></tr>
|
|
</table>
|
|
</td></tr></table></body>
|
|
</html>
|
|
"""
|
|
|
|
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']]]
|
|
|
|
|
|
class OutputTestCase(unittest.TestCase):
|
|
def normalize_calendar(self, s):
|
|
# Filters out locale dependent strings
|
|
def neitherspacenordigit(c):
|
|
return not c.isspace() and not c.isdigit()
|
|
|
|
lines = []
|
|
for line in s.splitlines(keepends=False):
|
|
# Drop texts, as they are locale dependent
|
|
if line and not filter(neitherspacenordigit, line):
|
|
lines.append(line)
|
|
return lines
|
|
|
|
def check_htmlcalendar_encoding(self, req, res):
|
|
cal = calendar.HTMLCalendar()
|
|
format_ = default_format.copy()
|
|
format_["encoding"] = req or 'utf-8'
|
|
output = cal.formatyearpage(2004, encoding=req)
|
|
self.assertEqual(
|
|
output,
|
|
result_2004_html.format(**format_).encode(res)
|
|
)
|
|
|
|
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(
|
|
calendar.TextCalendar().formatyear(2004),
|
|
result_2004_text
|
|
)
|
|
|
|
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):
|
|
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]
|
|
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):
|
|
self.assertEqual(
|
|
calendar.TextCalendar().formatweekheader(9),
|
|
' Monday Tuesday Wednesday Thursday '
|
|
' Friday Saturday Sunday '
|
|
)
|
|
|
|
def test_formatmonth(self):
|
|
self.assertEqual(
|
|
calendar.TextCalendar().formatmonth(2004, 1),
|
|
result_2004_01_text
|
|
)
|
|
|
|
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>'
|
|
)
|
|
|
|
def test_prweek(self):
|
|
with support.captured_stdout() as out:
|
|
week = [(1,0), (2,1), (3,2), (4,3), (5,4), (6,5), (7,6)]
|
|
calendar.TextCalendar().prweek(week, 1)
|
|
self.assertEqual(out.getvalue(), " 1 2 3 4 5 6 7")
|
|
|
|
def test_prmonth(self):
|
|
with support.captured_stdout() as out:
|
|
calendar.TextCalendar().prmonth(2004, 1)
|
|
self.assertEqual(out.getvalue(), result_2004_01_text)
|
|
|
|
def test_pryear(self):
|
|
with support.captured_stdout() as out:
|
|
calendar.TextCalendar().pryear(2004)
|
|
self.assertEqual(out.getvalue(), result_2004_text)
|
|
|
|
def test_format(self):
|
|
with support.captured_stdout() as out:
|
|
calendar.format(["1", "2", "3"], colwidth=3, spacing=1)
|
|
self.assertEqual(out.getvalue().strip(), "1 2 3")
|
|
|
|
class CalendarTestCase(unittest.TestCase):
|
|
def test_isleap(self):
|
|
# 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):
|
|
self.assertRaises(TypeError, calendar.setfirstweekday, 'flabber')
|
|
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)
|
|
|
|
def test_illegal_weekday_reported(self):
|
|
with self.assertRaisesRegex(calendar.IllegalWeekdayError, '123'):
|
|
calendar.setfirstweekday(123)
|
|
|
|
def test_enumerate_weekdays(self):
|
|
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)
|
|
|
|
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
|
|
self.assertEqual(len(set(value)), 7)
|
|
# verify it "acts like a sequence" in two forms of iteration
|
|
self.assertEqual(value[::-1], list(reversed(value)))
|
|
|
|
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
|
|
self.assertEqual(len(set(value)), 13)
|
|
# verify it "acts like a sequence" in two forms of iteration
|
|
self.assertEqual(value[::-1], list(reversed(value)))
|
|
|
|
def test_locale_calendars(self):
|
|
# ensure that Locale{Text,HTML}Calendar resets the locale properly
|
|
# (it is still not thread-safe though)
|
|
old_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)
|
|
try:
|
|
cal = calendar.LocaleTextCalendar(locale='')
|
|
local_weekday = cal.formatweekday(1, 10)
|
|
local_month = cal.formatmonthname(2010, 10, 10)
|
|
except locale.Error:
|
|
# cannot set the system default locale -- skip rest of test
|
|
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)
|
|
new_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)
|
|
self.assertEqual(old_october, new_october)
|
|
|
|
def test_itermonthdates(self):
|
|
# ensure itermonthdates doesn't overflow after datetime.MAXYEAR
|
|
# see #15421
|
|
list(calendar.Calendar().itermonthdates(datetime.MAXYEAR, 12))
|
|
|
|
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)
|
|
|
|
|
|
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))
|
|
for i in range(len(weeks)):
|
|
self.assertEqual(weeks[i], sum(day != 0 for day in cal[i]))
|
|
|
|
|
|
class MondayTestCase(MonthCalendarTestCase):
|
|
firstweekday = calendar.MONDAY
|
|
|
|
def test_february(self):
|
|
# A 28-day february starting on monday (7+7+7+7 days)
|
|
self.check_weeks(1999, 2, (7, 7, 7, 7))
|
|
|
|
# A 28-day february starting on tuesday (6+7+7+7+1 days)
|
|
self.check_weeks(2005, 2, (6, 7, 7, 7, 1))
|
|
|
|
# A 28-day february starting on sunday (1+7+7+7+6 days)
|
|
self.check_weeks(1987, 2, (1, 7, 7, 7, 6))
|
|
|
|
# A 29-day february starting on monday (7+7+7+7+1 days)
|
|
self.check_weeks(1988, 2, (7, 7, 7, 7, 1))
|
|
|
|
# A 29-day february starting on tuesday (6+7+7+7+2 days)
|
|
self.check_weeks(1972, 2, (6, 7, 7, 7, 2))
|
|
|
|
# A 29-day february starting on sunday (1+7+7+7+7 days)
|
|
self.check_weeks(2004, 2, (1, 7, 7, 7, 7))
|
|
|
|
def test_april(self):
|
|
# A 30-day april starting on monday (7+7+7+7+2 days)
|
|
self.check_weeks(1935, 4, (7, 7, 7, 7, 2))
|
|
|
|
# A 30-day april starting on tuesday (6+7+7+7+3 days)
|
|
self.check_weeks(1975, 4, (6, 7, 7, 7, 3))
|
|
|
|
# A 30-day april starting on sunday (1+7+7+7+7+1 days)
|
|
self.check_weeks(1945, 4, (1, 7, 7, 7, 7, 1))
|
|
|
|
# A 30-day april starting on saturday (2+7+7+7+7 days)
|
|
self.check_weeks(1995, 4, (2, 7, 7, 7, 7))
|
|
|
|
# A 30-day april starting on friday (3+7+7+7+6 days)
|
|
self.check_weeks(1994, 4, (3, 7, 7, 7, 6))
|
|
|
|
def test_december(self):
|
|
# A 31-day december starting on monday (7+7+7+7+3 days)
|
|
self.check_weeks(1980, 12, (7, 7, 7, 7, 3))
|
|
|
|
# A 31-day december starting on tuesday (6+7+7+7+4 days)
|
|
self.check_weeks(1987, 12, (6, 7, 7, 7, 4))
|
|
|
|
# A 31-day december starting on sunday (1+7+7+7+7+2 days)
|
|
self.check_weeks(1968, 12, (1, 7, 7, 7, 7, 2))
|
|
|
|
# A 31-day december starting on thursday (4+7+7+7+6 days)
|
|
self.check_weeks(1988, 12, (4, 7, 7, 7, 6))
|
|
|
|
# A 31-day december starting on friday (3+7+7+7+7 days)
|
|
self.check_weeks(2017, 12, (3, 7, 7, 7, 7))
|
|
|
|
# A 31-day december starting on saturday (2+7+7+7+7+1 days)
|
|
self.check_weeks(2068, 12, (2, 7, 7, 7, 7, 1))
|
|
|
|
|
|
class SundayTestCase(MonthCalendarTestCase):
|
|
firstweekday = calendar.SUNDAY
|
|
|
|
def test_february(self):
|
|
# A 28-day february starting on sunday (7+7+7+7 days)
|
|
self.check_weeks(2009, 2, (7, 7, 7, 7))
|
|
|
|
# A 28-day february starting on monday (6+7+7+7+1 days)
|
|
self.check_weeks(1999, 2, (6, 7, 7, 7, 1))
|
|
|
|
# A 28-day february starting on saturday (1+7+7+7+6 days)
|
|
self.check_weeks(1997, 2, (1, 7, 7, 7, 6))
|
|
|
|
# A 29-day february starting on sunday (7+7+7+7+1 days)
|
|
self.check_weeks(2004, 2, (7, 7, 7, 7, 1))
|
|
|
|
# A 29-day february starting on monday (6+7+7+7+2 days)
|
|
self.check_weeks(1960, 2, (6, 7, 7, 7, 2))
|
|
|
|
# A 29-day february starting on saturday (1+7+7+7+7 days)
|
|
self.check_weeks(1964, 2, (1, 7, 7, 7, 7))
|
|
|
|
def test_april(self):
|
|
# A 30-day april starting on sunday (7+7+7+7+2 days)
|
|
self.check_weeks(1923, 4, (7, 7, 7, 7, 2))
|
|
|
|
# A 30-day april starting on monday (6+7+7+7+3 days)
|
|
self.check_weeks(1918, 4, (6, 7, 7, 7, 3))
|
|
|
|
# A 30-day april starting on saturday (1+7+7+7+7+1 days)
|
|
self.check_weeks(1950, 4, (1, 7, 7, 7, 7, 1))
|
|
|
|
# A 30-day april starting on friday (2+7+7+7+7 days)
|
|
self.check_weeks(1960, 4, (2, 7, 7, 7, 7))
|
|
|
|
# A 30-day april starting on thursday (3+7+7+7+6 days)
|
|
self.check_weeks(1909, 4, (3, 7, 7, 7, 6))
|
|
|
|
def test_december(self):
|
|
# A 31-day december starting on sunday (7+7+7+7+3 days)
|
|
self.check_weeks(2080, 12, (7, 7, 7, 7, 3))
|
|
|
|
# A 31-day december starting on monday (6+7+7+7+4 days)
|
|
self.check_weeks(1941, 12, (6, 7, 7, 7, 4))
|
|
|
|
# A 31-day december starting on saturday (1+7+7+7+7+2 days)
|
|
self.check_weeks(1923, 12, (1, 7, 7, 7, 7, 2))
|
|
|
|
# A 31-day december starting on wednesday (4+7+7+7+6 days)
|
|
self.check_weeks(1948, 12, (4, 7, 7, 7, 6))
|
|
|
|
# A 31-day december starting on thursday (3+7+7+7+7 days)
|
|
self.check_weeks(1927, 12, (3, 7, 7, 7, 7))
|
|
|
|
# A 31-day december starting on friday (2+7+7+7+7+1 days)
|
|
self.check_weeks(1995, 12, (2, 7, 7, 7, 7, 1))
|
|
|
|
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))
|
|
|
|
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)
|
|
|
|
def test_illegal_month_reported(self):
|
|
with self.assertRaisesRegex(calendar.IllegalMonthError, '65'):
|
|
calendar.monthrange(2004, 65)
|
|
|
|
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)
|
|
|
|
|
|
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)
|
|
self.assertIn(b'usage:', stderr)
|
|
self.assertEqual(rc, 2)
|
|
|
|
def test_help(self):
|
|
stdout = self.run_ok('-h')
|
|
self.assertIn(b'usage:', stdout)
|
|
self.assertIn(b'calendar.py', stdout)
|
|
self.assertIn(b'--help', stdout)
|
|
|
|
def test_illegal_arguments(self):
|
|
self.assertFailure('-z')
|
|
self.assertFailure('spam')
|
|
self.assertFailure('2004', 'spam')
|
|
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')
|
|
self.assertEqual(stdout[:6], b'<?xml ')
|
|
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.format(**default_format).encode('ascii'))
|
|
|
|
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)
|
|
|
|
|
|
class MiscTestCase(unittest.TestCase):
|
|
def test__all__(self):
|
|
blacklist = {'mdays', 'January', 'February', 'EPOCH',
|
|
'MONDAY', 'TUESDAY', 'WEDNESDAY', 'THURSDAY', 'FRIDAY',
|
|
'SATURDAY', 'SUNDAY', 'different_locale', 'c',
|
|
'prweek', 'week', 'format', 'formatstring', 'main'}
|
|
support.check__all__(self, calendar, blacklist=blacklist)
|
|
|
|
|
|
class TestSubClassingCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
|
|
class CustomHTMLCal(calendar.HTMLCalendar):
|
|
cssclasses = [style + " text-nowrap" for style in
|
|
calendar.HTMLCalendar.cssclasses]
|
|
cssclasses_weekday_head = ["red", "blue", "green", "lilac",
|
|
"yellow", "orange", "pink"]
|
|
cssclass_month_head = "text-center month-head"
|
|
cssclass_month = "text-center month"
|
|
cssclass_year = "text-italic "
|
|
cssclass_year_head = "lead "
|
|
|
|
self.cal = CustomHTMLCal()
|
|
|
|
def test_formatmonthname(self):
|
|
self.assertIn('class="text-center month-head"',
|
|
self.cal.formatmonthname(2017, 5))
|
|
|
|
def test_formatmonth(self):
|
|
self.assertIn('class="text-center month"',
|
|
self.cal.formatmonth(2017, 5))
|
|
|
|
def test_formatweek(self):
|
|
weeks = self.cal.monthdays2calendar(2017, 5)
|
|
self.assertIn('class="wed text-nowrap"', self.cal.formatweek(weeks[0]))
|
|
|
|
def test_formatweek_head(self):
|
|
header = self.cal.formatweekheader()
|
|
for color in self.cal.cssclasses_weekday_head:
|
|
self.assertIn('<th class="%s">' % color, header)
|
|
|
|
def test_format_year(self):
|
|
self.assertIn(
|
|
('<table border="0" cellpadding="0" cellspacing="0" class="%s">' %
|
|
self.cal.cssclass_year), self.cal.formatyear(2017))
|
|
|
|
def test_format_year_head(self):
|
|
self.assertIn('<tr><th colspan="%d" class="%s">%s</th></tr>' % (
|
|
3, self.cal.cssclass_year_head, 2017), self.cal.formatyear(2017))
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|