Source Code for Module parsedatetime.tests.TestRanges

 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      targetStart, targetEnd, t_flag = result 
14      valueStart, valueEnd,  v_flag = check 
15   
16      t1_yr, t1_mth, t1_dy, t1_hr, t1_min, _, _, _, _ = targetStart 
17      v1_yr, v1_mth, v1_dy, v1_hr, v1_min, _, _, _, _ = valueStart 
18   
19      t2_yr, t2_mth, t2_dy, t2_hr, t2_min, _, _, _, _ = targetEnd 
20      v2_yr, v2_mth, v2_dy, v2_hr, v2_min, _, _, _, _ = valueEnd 
21   
22      return ((t1_yr == v1_yr) and (t1_mth == v1_mth) and (t1_dy == v1_dy) and (t1_hr == v1_hr) and 
23              (t1_min == v1_min) and (t2_yr == v2_yr) and (t2_mth == v2_mth) and (t2_dy == v2_dy) and 
24              (t2_hr == v2_hr) and (t2_min == v2_min) and (t_flag == v_flag)) 
25   
26   
27 -class test(unittest.TestCase): 
28   
29 -    def setUp(self): 
30          self.cal = pdt.Calendar() 
31          self.yr, self.mth, self.dy, self.hr, self.mn, self.sec, self.wd, self.yd, self.isdst = time.localtime() 
32   
33 -    def testTimes(self): 
34          start = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
35   
36          targetStart = datetime.datetime(self.yr, self.mth, self.dy, 14, 0, 0).timetuple() 
37          targetEnd   = datetime.datetime(self.yr, self.mth, self.dy, 17, 30, 0).timetuple() 
38   
39          self.assertTrue(_compareResults(self.cal.evalRanges("2 pm - 5:30 pm",          start), (targetStart, targetEnd, 2))) 
40          self.assertTrue(_compareResults(self.cal.evalRanges("2pm - 5:30pm",            start), (targetStart, targetEnd, 2))) 
41          self.assertTrue(_compareResults(self.cal.evalRanges("2:00:00 pm - 5:30:00 pm", start), (targetStart, targetEnd, 2))) 
42          self.assertTrue(_compareResults(self.cal.evalRanges("2 - 5:30pm",              start), (targetStart, targetEnd, 2))) 
43          self.assertTrue(_compareResults(self.cal.evalRanges("14:00 - 17:30",           start), (targetStart, targetEnd, 2))) 
44   
45          targetStart = datetime.datetime(self.yr, self.mth, self.dy, 10, 0, 0).timetuple() 
46          targetEnd   = datetime.datetime(self.yr, self.mth, self.dy, 13, 30, 0).timetuple() 
47   
48          self.assertTrue(_compareResults(self.cal.evalRanges("10AM - 1:30PM",            start), (targetStart, targetEnd, 2))) 
49          self.assertTrue(_compareResults(self.cal.evalRanges("10:00:00 am - 1:30:00 pm", start), (targetStart, targetEnd, 2))) 
50          self.assertTrue(_compareResults(self.cal.evalRanges("10:00 - 13:30",            start), (targetStart, targetEnd, 2))) 
51   
52          targetStart = datetime.datetime(self.yr, self.mth, self.dy, 15, 30, 0).timetuple() 
53          targetEnd   = datetime.datetime(self.yr, self.mth, self.dy, 17, 0, 0).timetuple() 
54   
55          self.assertTrue(_compareResults(self.cal.evalRanges("today 3:30-5PM", start), (targetStart, targetEnd, 2))) 
56   
57 -    def testDates(self): 
58          start = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
59   
60          targetStart = datetime.datetime(2006, 8, 29, self.hr, self.mn, self.sec).timetuple() 
61          targetEnd   = datetime.datetime(2006, 9, 2,self.hr, self.mn, self.sec).timetuple() 
62   
63          self.assertTrue(_compareResults(self.cal.evalRanges("August 29, 2006 - September 2, 2006", start), (targetStart, targetEnd, 1))) 
64          self.assertTrue(_compareResults(self.cal.evalRanges("August 29 - September 2, 2006",       start), (targetStart, targetEnd, 1))) 
65   
66          targetStart = datetime.datetime(2006, 8, 29, self.hr, self.mn, self.sec).timetuple() 
67          targetEnd   = datetime.datetime(2006, 9, 2, self.hr, self.mn, self.sec).timetuple() 
68   
69          self.assertTrue(_compareResults(self.cal.evalRanges("08/29/06 - 09/02/06", start), (targetStart, targetEnd, 1))) 
70   
71   
72      #def testSubRanges(self): 
73      #    start = datetime.datetime(self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple() 
74   
75      #    targetStart = datetime.datetime(2006, 8, 1, 9, 0, 0).timetuple() 
76      #    targetEnd   = datetime.datetime(2006, 8, 15, 9, 0, 0).timetuple() 
77   
78      #    self.assertTrue(_compareResults(self.cal.evalRanges("August 1-15, 2006", start), (targetStart, targetEnd, 1))) 
79   
80   
81  if __name__ == "__main__": 
82      unittest.main() 
83