Source Code for Module parsedatetime.tests.TestMultiple

 1   
 2  """ 
 3  Test parsing of strings with multiple chunks 
 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   
25 -    def setUp(self): 
26          self.cal = pdt.Calendar() 
27          self.yr, self.mth, self.dy, self.hr, self.mn, self.sec, self.wd, self.yd, self.isdst = time.localtime() 
28   
29 -    def testSimpleMultipleItems(self): 
30          s = datetime.datetime.now() 
31          t = self.cal.inc(s, year=3) + datetime.timedelta(days=5, weeks=2) 
32   
33          start  = s.timetuple() 
34          target = t.timetuple() 
35   
36          self.assertTrue(_compareResults(self.cal.parse('3 years 2 weeks 5 days', start), (target, 1))) 
37          self.assertTrue(_compareResults(self.cal.parse('3years 2weeks 5days',    start), (target, 1))) 
38   
39 -    def testMultipleItemsSingleCharUnits(self): 
40          s = datetime.datetime.now() 
41          t = self.cal.inc(s, year=3) + datetime.timedelta(days=5, weeks=2) 
42   
43          start  = s.timetuple() 
44          target = t.timetuple() 
45   
46          self.assertTrue(_compareResults(self.cal.parse('3 y 2 w 5 d', start), (target, 1))) 
47          self.assertTrue(_compareResults(self.cal.parse('3y 2w 5d',    start), (target, 1))) 
48   
49          t      = self.cal.inc(s, year=3) + datetime.timedelta(hours=5, minutes=50) 
50          target = t.timetuple() 
51   
52          self.assertTrue(_compareResults(self.cal.parse('3y 5h 50m', start), (target, 3))) 
53   
54 -    def testMultipleItemsWithPunctuation(self): 
55          s = datetime.datetime.now() 
56          t = self.cal.inc(s, year=3) + datetime.timedelta(days=5, weeks=2) 
57   
58          start  = s.timetuple() 
59          target = t.timetuple() 
60   
61          self.assertTrue(_compareResults(self.cal.parse('3 years, 2 weeks, 5 days',    start), (target, 1))) 
62          self.assertTrue(_compareResults(self.cal.parse('3 years, 2 weeks and 5 days', start), (target, 1))) 
63          self.assertTrue(_compareResults(self.cal.parse('3y, 2w, 5d ',                 start), (target, 1))) 
64   
65 -    def testUnixATStyle(self): 
66          s = datetime.datetime.now() 
67          t = s + datetime.timedelta(days=3) 
68   
69          t = t.replace(hour=16, minute=0, second=0) 
70   
71          start  = s.timetuple() 
72          target = t.timetuple() 
73   
74          self.assertTrue(_compareResults(self.cal.parse('4pm + 3 days', start), (target, 3))) 
75          self.assertTrue(_compareResults(self.cal.parse('4pm +3 days',  start), (target, 3))) 
76   
77 -    def testUnixATStyleNegative(self): 
78          s = datetime.datetime.now() 
79          t = s + datetime.timedelta(days=-3) 
80   
81          t = t.replace(hour=16, minute=0, second=0) 
82   
83          start  = s.timetuple() 
84          target = t.timetuple() 
85   
86          self.assertTrue(_compareResults(self.cal.parse('4pm - 3 days', start), (target, 3))) 
87          self.assertTrue(_compareResults(self.cal.parse('4pm -3 days',  start), (target, 3))) 
88   
89   
90  if __name__ == "__main__": 
91      unittest.main() 
92