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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
# -*- coding: utf-8 -*-
"""
Test Calendar.Inc() routine
"""
from __future__ import unicode_literals
import sys
import time
import datetime
import parsedatetime as pdt
from . import utils
if sys.version_info < (2, 7):
import unittest2 as unittest
else:
import unittest
class test(unittest.TestCase):
@utils.assertEqualWithComparator
def assertExpectedResult(self, result, check, **kwargs):
return utils.compareResultByTimeTuplesAndFlags((result, 1),
(check, 1), **kwargs)
def setUp(self):
self.cal = pdt.Calendar()
(self.yr, self.mth, self.dy, self.hr,
self.mn, self.sec, self.wd, self.yd, self.isdst) = time.localtime()
def testIncMonths(self):
s = datetime.datetime(2006, 1, 1, 12, 0, 0)
t = datetime.datetime(2006, 2, 1, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, month=1).timetuple(), t.timetuple())
s = datetime.datetime(2006, 12, 1, 12, 0, 0)
t = datetime.datetime(2007, 1, 1, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, month=1).timetuple(), t.timetuple())
# leap year, Feb 1
s = datetime.datetime(2008, 2, 1, 12, 0, 0)
t = datetime.datetime(2008, 3, 1, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, month=1).timetuple(), t.timetuple())
# leap year, Feb 29
s = datetime.datetime(2008, 2, 29, 12, 0, 0)
t = datetime.datetime(2008, 3, 29, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, month=1).timetuple(), t.timetuple())
s = datetime.datetime(2006, 1, 1, 12, 0, 0)
t = datetime.datetime(2005, 12, 1, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, month=-1).timetuple(), t.timetuple())
# End of month Jan 31 to Feb - Febuary only has 28 days
s = datetime.datetime(2006, 1, 31, 12, 0, 0)
t = datetime.datetime(2006, 2, 28, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, month=1).timetuple(), t.timetuple())
# walk thru months and make sure month increment doesn't set the day
# to be past the last day of the new month
# think Jan transition to Feb - 31 days to 28 days
for m in range(1, 11):
d = self.cal.ptc.daysInMonth(m, 2006)
s = datetime.datetime(2006, m, d, 12, 0, 0)
if d > self.cal.ptc.daysInMonth(m + 1, 2006):
d = self.cal.ptc.daysInMonth(m + 1, 2006)
t = datetime.datetime(2006, m + 1, d, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, month=1).timetuple(), t.timetuple())
def testIncYears(self):
s = datetime.datetime(2006, 1, 1, 12, 0, 0)
t = datetime.datetime(2007, 1, 1, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, year=1).timetuple(), t.timetuple())
s = datetime.datetime(2006, 1, 1, 12, 0, 0)
t = datetime.datetime(2008, 1, 1, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, year=2).timetuple(), t.timetuple())
s = datetime.datetime(2006, 12, 31, 12, 0, 0)
t = datetime.datetime(2007, 12, 31, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, year=1).timetuple(), t.timetuple())
s = datetime.datetime(2006, 12, 31, 12, 0, 0)
t = datetime.datetime(2005, 12, 31, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, year=-1).timetuple(), t.timetuple())
s = datetime.datetime(2008, 3, 1, 12, 0, 0)
t = datetime.datetime(2009, 3, 1, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, year=1).timetuple(), t.timetuple())
s = datetime.datetime(2008, 3, 1, 12, 0, 0)
t = datetime.datetime(2007, 3, 1, 12, 0, 0)
self.assertExpectedResult(
self.cal.inc(s, year=-1).timetuple(), t.timetuple())
if __name__ == "__main__":
unittest.main()
|
