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 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164
|
# -*- coding: utf-8 -*-
"""
Test parsing of complex date and times
"""
from __future__ import unicode_literals
import sys
import time
from datetime 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, check, **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 testDate3ConfusedHourAndYear(self):
start = datetime(
self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple()
self.assertExpectedResult(
self.cal.parse('Aug 05, 2014 4:15 AM'),
(datetime(2014, 8, 5, 4, 15, 0).timetuple(), 3))
self.assertExpectedResult(
self.cal.parse('Aug 05, 2003 3:15 AM'),
(datetime(2003, 8, 5, 3, 15, 0).timetuple(), 3))
self.assertExpectedResult(
self.cal.parse('Aug 05, 2003 03:15 AM'),
(datetime(2003, 8, 5, 3, 15, 0).timetuple(), 3))
self.assertExpectedResult(
self.cal.parse('June 30th 12PM', start),
(datetime(self.yr
if datetime.now() < datetime(self.yr, 6, 30, 12)
else self.yr + 1,
6, 30, 12, 0, 0).timetuple(), 3))
self.assertExpectedResult(
self.cal.parse('June 30th 12:00', start),
(datetime(self.yr
if datetime.now() < datetime(self.yr, 6, 30, 12)
else self.yr + 1,
6, 30, 12, 0, 0).timetuple(), 3))
self.assertExpectedResult(
self.cal.parse('December 30th 23PM', start),
(datetime(self.yr
if datetime.now() < datetime(self.yr, 12, 30, 23)
else self.yr + 1,
12, 30, 23, 0, 0).timetuple(), 3))
self.assertExpectedResult(
self.cal.parse('December 30th 23:02', start),
(datetime(self.yr
if datetime.now() < datetime(self.yr, 12, 30, 23, 2)
else self.yr + 1,
12, 30, 23, 2, 0).timetuple(), 3))
def testDates(self):
start = datetime(
self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple()
target = datetime(2006, 8, 25, 17, 0, 0).timetuple()
self.assertExpectedResult(
self.cal.parse('08/25/2006 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5pm on 08.25.2006', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5pm August 25, 2006', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5pm August 25th, 2006', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5pm 25 August, 2006', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5pm 25th August, 2006', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('Aug 25, 2006 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('Aug 25th, 2006 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('25 Aug, 2006 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('25th Aug 2006, 5pm', start), (target, 3))
if self.mth > 8 or (self.mth == 8 and self.dy > 5):
target = datetime(self.yr + 1, 8, 5, 17, 0, 0).timetuple()
else:
target = datetime(self.yr, 8, 5, 17, 0, 0).timetuple()
self.assertExpectedResult(
self.cal.parse('8/5 at 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5pm 8.5', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('08/05 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('August 5 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5pm Aug 05', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('Aug 05 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('Aug 05th 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5 August 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5th August 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('5pm 05 Aug', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('05 Aug 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('05th Aug 5pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('August 5th 5pm', start), (target, 3))
if self.mth > 8 or (self.mth == 8 and self.dy > 5):
target = datetime(self.yr + 1, 8, 5, 12, 0, 0).timetuple()
else:
target = datetime(self.yr, 8, 5, 12, 0, 0).timetuple()
self.assertExpectedResult(
self.cal.parse('August 5th 12:00', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('August 5th 12pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('August 5th 12:00pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('August 5th 12 pm', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('August 5th 12:00 pm', start), (target, 3))
if self.mth > 8 or (self.mth == 8 and self.dy > 22):
target = datetime(
self.yr + 1, 8, 22, 3, 26, 0).timetuple()
else:
target = datetime(self.yr, 8, 22, 3, 26, 0).timetuple()
self.assertExpectedResult(
self.cal.parse('August 22nd 3:26', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('August 22nd 3:26am', start), (target, 3))
self.assertExpectedResult(
self.cal.parse('August 22nd 3:26 am', start), (target, 3))
def testDatesWithDay(self):
start = datetime(
self.yr, self.mth, self.dy, self.hr, self.mn, self.sec).timetuple()
target = datetime(2016, 8, 23, 17, 0, 0).timetuple()
self.assertExpectedResult(
self.cal.parse('tuesday august 23nd 2016 at 5pm', start), (target, 3))
if __name__ == "__main__":
unittest.main()
|