Source Code for Module parsedatetime.tests.TestSimpleDateTimes

  1   
  2  """ 
  3  Test parsing of simple date and times 
  4  """ 
  5   
  6  import unittest, time, datetime 
  7  import parsedatetime as pdt 
  8   
  9   
 10    # a special compare function is used to allow us to ignore the seconds as 
 11    # the running of the test could cross a minute boundary 
 12 -def _compareResults(result, check): 
 13      target, t_flag = result 
 14      value,  v_flag = check 
 15   
 16      t_yr, t_mth, t_dy, t_hr, t_min, _, _, _, _ = target 
 17      v_yr, v_mth, v_dy, v_hr, v_min, _, _, _, _ = value 
 18   
 19      return ((t_yr == v_yr) and (t_mth == v_mth) and (t_dy == v_dy) and 
 20              (t_hr == v_hr) and (t_min == v_min)) and (t_flag == v_flag) 
 21   
 22   
 23 -class test(unittest.TestCase): 
 24 -    def setUp(self): 
 25          self.cal = pdt.Calendar() 
 26          self.yr, self.mth, self.dy, self.hr, self.mn, self.sec, self.wd, self.yd, self.isdst = time.localtime() 
 27   
 28 -    def testDays(self): 
 29          s = datetime.datetime.now() 
 30          t = s + datetime.timedelta(days=1) 
 31   
 32          start  = s.timetuple() 
 33          target = t.timetuple() 
 34   
 35          d = self.wd + 1 
 36   
 37          if d > 6: 
 38              d = 0 
 39   
 40          day = self.cal.ptc.Weekdays[d] 
 41   
 42          self.assertTrue(_compareResults(self.cal.parse(day, start), (target, 1))) 
 43   
 44          t = s + datetime.timedelta(days=6) 
 45   
 46          target = t.timetuple() 
 47   
 48          d = self.wd - 1 
 49   
 50          if d < 0: 
 51              d = 6 
 52   
 53          day = self.cal.ptc.Weekdays[d] 
 54   
 55          self.assertTrue(_compareResults(self.cal.parse(day, start), (target, 1))) 
 56   
 57 -    def testTimes(self): 
 58          start  = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
 59          target = datetime.datetime(self.yr, self.mth, self.dy, 23, 0, 0).timetuple() 
 60   
 61          self.assertTrue(_compareResults(self.cal.parse('11:00:00 PM',   start), (target, 2))) 
 62          self.assertTrue(_compareResults(self.cal.parse('11:00 PM',      start), (target, 2))) 
 63          self.assertTrue(_compareResults(self.cal.parse('11 PM',         start), (target, 2))) 
 64          self.assertTrue(_compareResults(self.cal.parse('11PM',          start), (target, 2))) 
 65          self.assertTrue(_compareResults(self.cal.parse('2300',          start), (target, 2))) 
 66          self.assertTrue(_compareResults(self.cal.parse('23:00',         start), (target, 2))) 
 67          self.assertTrue(_compareResults(self.cal.parse('11p',           start), (target, 2))) 
 68          self.assertTrue(_compareResults(self.cal.parse('11pm',          start), (target, 2))) 
 69          self.assertTrue(_compareResults(self.cal.parse('11:00:00 P.M.', start), (target, 2))) 
 70          self.assertTrue(_compareResults(self.cal.parse('11:00 P.M.',    start), (target, 2))) 
 71          self.assertTrue(_compareResults(self.cal.parse('11 P.M.',       start), (target, 2))) 
 72          self.assertTrue(_compareResults(self.cal.parse('11P.M.',        start), (target, 2))) 
 73          self.assertTrue(_compareResults(self.cal.parse('11p.m.',        start), (target, 2))) 
 74          self.assertTrue(_compareResults(self.cal.parse('11 p.m.',       start), (target, 2))) 
 75   
 76          target = datetime.datetime(self.yr, self.mth, self.dy, 11, 0, 0).timetuple() 
 77   
 78          self.assertTrue(_compareResults(self.cal.parse('11:00:00 AM', start), (target, 2))) 
 79          self.assertTrue(_compareResults(self.cal.parse('11:00 AM',    start), (target, 2))) 
 80          self.assertTrue(_compareResults(self.cal.parse('11 AM',       start), (target, 2))) 
 81          self.assertTrue(_compareResults(self.cal.parse('11AM',        start), (target, 2))) 
 82          self.assertTrue(_compareResults(self.cal.parse('1100',        start), (target, 2))) 
 83          self.assertTrue(_compareResults(self.cal.parse('11:00',       start), (target, 2))) 
 84          self.assertTrue(_compareResults(self.cal.parse('11a',         start), (target, 2))) 
 85          self.assertTrue(_compareResults(self.cal.parse('11am',        start), (target, 2))) 
 86          self.assertTrue(_compareResults(self.cal.parse('11:00:00 A.M.', start), (target, 2))) 
 87          self.assertTrue(_compareResults(self.cal.parse('11:00 A.M.',    start), (target, 2))) 
 88          self.assertTrue(_compareResults(self.cal.parse('11 A.M.',       start), (target, 2))) 
 89          self.assertTrue(_compareResults(self.cal.parse('11A.M.',        start), (target, 2))) 
 90          self.assertTrue(_compareResults(self.cal.parse('11a.m.',        start), (target, 2))) 
 91          self.assertTrue(_compareResults(self.cal.parse('11 a.m.',       start), (target, 2))) 
 92   
 93          target = datetime.datetime(self.yr, self.mth, self.dy, 7, 30, 0).timetuple() 
 94   
 95          self.assertTrue(_compareResults(self.cal.parse('730',  start), (target, 2))) 
 96          self.assertTrue(_compareResults(self.cal.parse('0730', start), (target, 2))) 
 97   
 98          target = datetime.datetime(self.yr, self.mth, self.dy, 17, 30, 0).timetuple() 
 99   
100          self.assertTrue(_compareResults(self.cal.parse('1730',   start), (target, 2))) 
101          self.assertTrue(_compareResults(self.cal.parse('173000', start), (target, 2))) 
102   
103 -    def testDates(self): 
104          start  = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
105          target = datetime.datetime(2006, 8, 25,  self.hr, self.mn, self.sec).timetuple() 
106   
107          self.assertTrue(_compareResults(self.cal.parse('08/25/2006',      start), (target, 1))) 
108          self.assertTrue(_compareResults(self.cal.parse('08.25.2006',      start), (target, 1))) 
109          self.assertTrue(_compareResults(self.cal.parse('8/25/06',         start), (target, 1))) 
110          self.assertTrue(_compareResults(self.cal.parse('August 25, 2006', start), (target, 1))) 
111          self.assertTrue(_compareResults(self.cal.parse('Aug 25, 2006',    start), (target, 1))) 
112          self.assertTrue(_compareResults(self.cal.parse('Aug. 25, 2006',   start), (target, 1))) 
113          self.assertTrue(_compareResults(self.cal.parse('August 25 2006',  start), (target, 1))) 
114          self.assertTrue(_compareResults(self.cal.parse('Aug 25 2006',     start), (target, 1))) 
115          self.assertTrue(_compareResults(self.cal.parse('Aug. 25 2006',    start), (target, 1))) 
116          self.assertTrue(_compareResults(self.cal.parse('25 August 2006',  start), (target, 1))) 
117          self.assertTrue(_compareResults(self.cal.parse('25 Aug 2006',     start), (target, 1))) 
118   
119          if self.mth > 8 or (self.mth == 8 and self.dy > 25): 
120              target = datetime.datetime(self.yr + 1, 8, 25,  self.hr, self.mn, self.sec).timetuple() 
121          else: 
122              target = datetime.datetime(self.yr, 8, 25,  self.hr, self.mn, self.sec).timetuple() 
123   
124          self.assertTrue(_compareResults(self.cal.parse('8/25',      start), (target, 1))) 
125          self.assertTrue(_compareResults(self.cal.parse('8.25',      start), (target, 1))) 
126          self.assertTrue(_compareResults(self.cal.parse('08/25',     start), (target, 1))) 
127          self.assertTrue(_compareResults(self.cal.parse('August 25', start), (target, 1))) 
128          self.assertTrue(_compareResults(self.cal.parse('Aug 25',    start), (target, 1))) 
129          self.assertTrue(_compareResults(self.cal.parse('Aug. 25',   start), (target, 1))) 
130   
131          # added test to ensure 4-digit year is recognized in the absence of day 
132          target = datetime.datetime(2013, 8, 1,  self.hr, self.mn, self.sec).timetuple() 
133          self.assertTrue(_compareResults(self.cal.parse('Aug. 2013',   start), (target, 1))) 
134   
135 -    def testLeapDays(self): 
136          start  = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
137          target = datetime.datetime(2000, 2, 29,  self.hr, self.mn, self.sec).timetuple() 
138   
139          self.assertTrue(_compareResults(self.cal.parse('02/29/2000', start), (target, 1))) 
140   
141          target = datetime.datetime(2004, 2, 29,  self.hr, self.mn, self.sec).timetuple() 
142   
143          self.assertTrue(_compareResults(self.cal.parse('02/29/2004', start), (target, 1))) 
144   
145          target = datetime.datetime(2008, 2, 29,  self.hr, self.mn, self.sec).timetuple() 
146   
147          self.assertTrue(_compareResults(self.cal.parse('02/29/2008', start), (target, 1))) 
148   
149          target = datetime.datetime(2012, 2, 29,  self.hr, self.mn, self.sec).timetuple() 
150   
151          self.assertTrue(_compareResults(self.cal.parse('02/29/2012', start), (target, 1))) 
152   
153          dNormal = (31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31) 
154          dLeap   = (31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31) 
155   
156          for i in range(1,12): 
157              self.assertTrue(self.cal.ptc.daysInMonth(i, 1999), dNormal[i - 1]) 
158              self.assertTrue(self.cal.ptc.daysInMonth(i, 2000), dLeap[i - 1]) 
159              self.assertTrue(self.cal.ptc.daysInMonth(i, 2001), dNormal[i - 1]) 
160              self.assertTrue(self.cal.ptc.daysInMonth(i, 2002), dNormal[i - 1]) 
161              self.assertTrue(self.cal.ptc.daysInMonth(i, 2003), dNormal[i - 1]) 
162              self.assertTrue(self.cal.ptc.daysInMonth(i, 2004), dLeap[i - 1]) 
163              self.assertTrue(self.cal.ptc.daysInMonth(i, 2005), dNormal[i - 1]) 
164   
165 -    def testDaySuffixes(self): 
166          start  = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
167          target = datetime.datetime(2008, 8, 22,  self.hr, self.mn, self.sec).timetuple() 
168   
169          self.assertTrue(_compareResults(self.cal.parse('August 22nd, 2008', start), (target, 1))) 
170          self.assertTrue(_compareResults(self.cal.parse('Aug 22nd, 2008',    start), (target, 1))) 
171          self.assertTrue(_compareResults(self.cal.parse('Aug. 22nd, 2008',   start), (target, 1))) 
172          self.assertTrue(_compareResults(self.cal.parse('August 22nd 2008',  start), (target, 1))) 
173          self.assertTrue(_compareResults(self.cal.parse('Aug 22nd 2008',     start), (target, 1))) 
174          self.assertTrue(_compareResults(self.cal.parse('Aug. 22nd 2008',    start), (target, 1))) 
175          self.assertTrue(_compareResults(self.cal.parse('22nd August 2008',  start), (target, 1))) 
176          self.assertTrue(_compareResults(self.cal.parse('22nd Aug 2008',     start), (target, 1))) 
177   
178          target = datetime.datetime(1949, 12, 31,  self.hr, self.mn, self.sec).timetuple() 
179   
180          self.assertTrue(_compareResults(self.cal.parse('December 31st, 1949', start), (target, 1))) 
181          self.assertTrue(_compareResults(self.cal.parse('Dec 31st, 1949',      start), (target, 1))) 
182          self.assertTrue(_compareResults(self.cal.parse('December 31st 1949',  start), (target, 1))) 
183          self.assertTrue(_compareResults(self.cal.parse('Dec 31st 1949',       start), (target, 1))) 
184          self.assertTrue(_compareResults(self.cal.parse('31st December 1949',  start), (target, 1))) 
185          self.assertTrue(_compareResults(self.cal.parse('31st Dec 1949',       start), (target, 1))) 
186   
187          target = datetime.datetime(2008, 8, 23,  self.hr, self.mn, self.sec).timetuple() 
188   
189          self.assertTrue(_compareResults(self.cal.parse('August 23rd, 2008', start), (target, 1))) 
190          self.assertTrue(_compareResults(self.cal.parse('Aug 23rd, 2008',    start), (target, 1))) 
191          self.assertTrue(_compareResults(self.cal.parse('Aug. 23rd, 2008',   start), (target, 1))) 
192          self.assertTrue(_compareResults(self.cal.parse('August 23rd 2008',  start), (target, 1))) 
193          self.assertTrue(_compareResults(self.cal.parse('Aug 23rd 2008',     start), (target, 1))) 
194          self.assertTrue(_compareResults(self.cal.parse('Aug. 23rd 2008',    start), (target, 1))) 
195   
196          target = datetime.datetime(2008, 8, 25,  self.hr, self.mn, self.sec).timetuple() 
197   
198          self.assertTrue(_compareResults(self.cal.parse('August 25th, 2008', start), (target, 1))) 
199          self.assertTrue(_compareResults(self.cal.parse('Aug 25th, 2008',    start), (target, 1))) 
200          self.assertTrue(_compareResults(self.cal.parse('Aug. 25th, 2008',   start), (target, 1))) 
201          self.assertTrue(_compareResults(self.cal.parse('August 25th 2008',  start), (target, 1))) 
202          self.assertTrue(_compareResults(self.cal.parse('Aug 25th 2008',     start), (target, 1))) 
203          self.assertTrue(_compareResults(self.cal.parse('Aug. 25th 2008',    start), (target, 1))) 
204   
205 -    def testSpecialTimes(self): 
206          start  = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
207          target = datetime.datetime(self.yr, self.mth, self.dy, 6, 0, 0).timetuple() 
208   
209          self.assertTrue(_compareResults(self.cal.parse('morning', start), (target, 2))) 
210   
211          target = datetime.datetime(self.yr, self.mth, self.dy, 8, 0, 0).timetuple() 
212   
213          self.assertTrue(_compareResults(self.cal.parse('breakfast', start), (target, 2))) 
214   
215          target = datetime.datetime(self.yr, self.mth, self.dy, 12, 0, 0).timetuple() 
216   
217          self.assertTrue(_compareResults(self.cal.parse('lunch', start), (target, 2))) 
218   
219          target = datetime.datetime(self.yr, self.mth, self.dy, 18, 0, 0).timetuple() 
220   
221          self.assertTrue(_compareResults(self.cal.parse('evening', start), (target, 2))) 
222   
223          target = datetime.datetime(self.yr, self.mth, self.dy, 19,0, 0).timetuple() 
224   
225          self.assertTrue(_compareResults(self.cal.parse('dinner', start), (target, 2))) 
226   
227          target = datetime.datetime(self.yr, self.mth, self.dy, 21, 0, 0).timetuple() 
228   
229          self.assertTrue(_compareResults(self.cal.parse('night',   start), (target, 2))) 
230          self.assertTrue(_compareResults(self.cal.parse('tonight', start), (target, 2))) 
231   
232 -    def testMidnight(self): 
233          start  = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
234          target = datetime.datetime(self.yr, self.mth, self.dy, 0, 0, 0).timetuple() 
235   
236          self.assertTrue(_compareResults(self.cal.parse('midnight',      start), (target, 2))) 
237          self.assertTrue(_compareResults(self.cal.parse('12:00:00 AM',   start), (target, 2))) 
238          self.assertTrue(_compareResults(self.cal.parse('12:00 AM',      start), (target, 2))) 
239          self.assertTrue(_compareResults(self.cal.parse('12 AM',         start), (target, 2))) 
240          self.assertTrue(_compareResults(self.cal.parse('12AM',          start), (target, 2))) 
241          self.assertTrue(_compareResults(self.cal.parse('12am',          start), (target, 2))) 
242          self.assertTrue(_compareResults(self.cal.parse('12a',           start), (target, 2))) 
243          self.assertTrue(_compareResults(self.cal.parse('0000',          start), (target, 2))) 
244          self.assertTrue(_compareResults(self.cal.parse('00:00',         start), (target, 2))) 
245          self.assertTrue(_compareResults(self.cal.parse('12:00:00 A.M.', start), (target, 2))) 
246          self.assertTrue(_compareResults(self.cal.parse('12:00 A.M.',    start), (target, 2))) 
247          self.assertTrue(_compareResults(self.cal.parse('12 A.M.',       start), (target, 2))) 
248          self.assertTrue(_compareResults(self.cal.parse('12A.M.',        start), (target, 2))) 
249          self.assertTrue(_compareResults(self.cal.parse('12a.m.',        start), (target, 2))) 
250   
251 -    def testNoon(self): 
252          start  = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
253          target = datetime.datetime(self.yr, self.mth, self.dy, 12, 0, 0).timetuple() 
254   
255          self.assertTrue(_compareResults(self.cal.parse('noon',          start), (target, 2))) 
256          self.assertTrue(_compareResults(self.cal.parse('12:00:00 PM',   start), (target, 2))) 
257          self.assertTrue(_compareResults(self.cal.parse('12:00 PM',      start), (target, 2))) 
258          self.assertTrue(_compareResults(self.cal.parse('12 PM',         start), (target, 2))) 
259          self.assertTrue(_compareResults(self.cal.parse('12PM',          start), (target, 2))) 
260          self.assertTrue(_compareResults(self.cal.parse('12pm',          start), (target, 2))) 
261          self.assertTrue(_compareResults(self.cal.parse('12p',           start), (target, 2))) 
262          self.assertTrue(_compareResults(self.cal.parse('1200',          start), (target, 2))) 
263          self.assertTrue(_compareResults(self.cal.parse('12:00',         start), (target, 2))) 
264          self.assertTrue(_compareResults(self.cal.parse('12:00:00 P.M.', start), (target, 2))) 
265          self.assertTrue(_compareResults(self.cal.parse('12:00 P.M.',    start), (target, 2))) 
266          self.assertTrue(_compareResults(self.cal.parse('12 P.M.',       start), (target, 2))) 
267          self.assertTrue(_compareResults(self.cal.parse('12P.M.',        start), (target, 2))) 
268          self.assertTrue(_compareResults(self.cal.parse('12p.m.',        start), (target, 2))) 
269   
270      # def testMonths(self): 
271      # 
272      #     start  = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
273      # 
274      #     target = datetime.datetime(self.yr, self.mth, self.dy, 12, 0, 0).timetuple() 
275      # 
276      #     self.assertTrue(_compareResults(self.cal.parse('jun',        start), (target, 2))) 
277      #     self.assertTrue(_compareResults(self.cal.parse('12:00:00 PM', start), (target, 2))) 
278      #     self.assertTrue(_compareResults(self.cal.parse('12:00 PM',    start), (target, 2))) 
279      #     self.assertTrue(_compareResults(self.cal.parse('12 PM',       start), (target, 2))) 
280      #     self.assertTrue(_compareResults(self.cal.parse('12PM',        start), (target, 2))) 
281      #     self.assertTrue(_compareResults(self.cal.parse('12pm',        start), (target, 2))) 
282      #     self.assertTrue(_compareResults(self.cal.parse('12p',         start), (target, 2))) 
283      #     self.assertTrue(_compareResults(self.cal.parse('1200',        start), (target, 2))) 
284      #     self.assertTrue(_compareResults(self.cal.parse('12:00',       start), (target, 2))) 
285   
286  if __name__ == "__main__": 
287      unittest.main() 
288