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
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
|
from whoosh.qparser.dateparse import *
basedate = datetime(2010, 9, 20, 15, 16, 6, 454000)
english = English()
def assert_adatetime(at, **kwargs):
assert at.__class__ is adatetime
for key in adatetime.units:
val = getattr(at, key)
target = kwargs.get(key)
assert val == target, "at.%s=%r not %r in %r" % (key, val, target, at)
def assert_timespan(ts, sargs, eargs):
assert_adatetime(ts.start, **sargs)
def assert_unamb(ts, **kwargs):
assert_unamb_span(ts, kwargs, kwargs)
def assert_unamb_span(ts, sargs, eargs):
startdt = adatetime(**sargs).floor()
enddt = adatetime(**eargs).ceil()
assert ts.start == startdt, "start %s != %s" % (ts.start, startdt)
assert ts.end == enddt, "end %s != %s" % (ts.end, enddt)
def assert_datespan(ts, startdate, enddate):
assert ts.__class__ is timespan
assert ts.start == startdate
assert ts.end == enddate
#
def test_simple_dateparse(t=english.simple):
assert_adatetime(t.date_from("2005", basedate), year=2005)
assert_adatetime(t.date_from("200505", basedate), year=2005, month=5)
assert_adatetime(t.date_from("20050510", basedate), year=2005, month=5,
day=10)
assert_adatetime(t.date_from("2005051001", basedate),
year=2005, month=5, day=10, hour=1)
assert_adatetime(t.date_from("200505100108", basedate),
year=2005, month=5, day=10, hour=1, minute=8)
assert_adatetime(t.date_from("20050510010835", basedate),
year=2005, month=5, day=10, hour=1, minute=8, second=35)
assert_adatetime(t.date_from("2005-05", basedate),
year=2005, month=5)
assert_adatetime(t.date_from("2005 05 10", basedate),
year=2005, month=5, day=10)
assert_adatetime(t.date_from("2005.05.10.01", basedate),
year=2005, month=5, day=10, hour=1)
assert_adatetime(t.date_from("2005/05/10 01:08", basedate),
year=2005, month=5, day=10, hour=1, minute=8)
assert_adatetime(t.date_from("2005.05.10 01:08:35", basedate),
year=2005, month=5, day=10, hour=1, minute=8, second=35)
assert t.date_from("2005 02 31", basedate) is None
assert t.date_from("2005-13-32", basedate) is None
def test_time(t=english.time):
assert_adatetime(t.date_from("13:05", basedate), hour=13, minute=5)
assert t.date_from("28:91", basedate) is None
assert_adatetime(t.date_from("3pm", basedate), hour=15)
assert_adatetime(t.date_from("3 pm", basedate), hour=15)
assert_adatetime(t.date_from("10pm", basedate), hour=22)
assert_adatetime(t.date_from("10 pm", basedate), hour=22)
assert_adatetime(t.date_from("3am", basedate), hour=3)
assert_adatetime(t.date_from("3:15 am", basedate), hour=3, minute=15)
assert_adatetime(t.date_from("5:10pm", basedate), hour=17, minute=10)
assert_adatetime(t.date_from("12:45am", basedate), hour=0, minute=45)
assert_adatetime(t.date_from("12:45pm", basedate), hour=12, minute=45)
assert_adatetime(t.date_from("5:45:05 pm", basedate),
hour=17, minute=45, second=5)
assert_adatetime(t.date_from("noon", basedate),
hour=12, minute=0, second=0, microsecond=0)
assert_adatetime(t.date_from("midnight", basedate),
hour=0, minute=0, second=0, microsecond=0)
assert t.date_from("15 am", basedate) is None
assert t.date_from("24:00", basedate) is None
assert t.date_from("12:65", basedate) is None
def test_dmy(d=english.dmy):
assert_adatetime(d.date_from("25 may 2011", basedate),
year=2011, month=5, day=25)
assert_adatetime(d.date_from("may 2 2011", basedate),
year=2011, month=5, day=2)
assert_adatetime(d.date_from("2011 25 may", basedate),
year=2011, month=5, day=25)
assert_adatetime(d.date_from("2011 may 5", basedate),
year=2011, month=5, day=5)
assert_adatetime(d.date_from("apr", basedate), month=4)
assert_adatetime(d.date_from("september", basedate), month=9)
assert_adatetime(d.date_from("2001", basedate), year=2001)
assert_adatetime(d.date_from("july 2525", basedate), year=2525, month=7)
assert_adatetime(d.date_from("nov 30", basedate), month=11, day=30)
assert d.date_from("25 2525", basedate) is None
assert_adatetime(d.date_from("25 may, 2011", basedate),
year=2011, month=5, day=25)
assert_adatetime(d.date_from("may 2nd, 2011", basedate),
year=2011, month=5, day=2)
assert_adatetime(d.date_from("2011, 25 may", basedate),
year=2011, month=5, day=25)
assert_adatetime(d.date_from("2011, may 5th", basedate),
year=2011, month=5, day=5)
assert_adatetime(d.date_from("today", basedate),
year=2010, month=9, day=20)
assert_adatetime(d.date_from("tomorrow", basedate),
year=2010, month=9, day=21)
assert_adatetime(d.date_from("yesterday", basedate),
year=2010, month=9, day=19)
assert_adatetime(d.date_from("this month", basedate), year=2010, month=9)
assert_adatetime(d.date_from("this year", basedate), year=2010)
assert d.date_from("now", basedate) == basedate
def test_plustime(rt=english.plusdate):
assert rt.date_from("+1hr", basedate) == basedate + timedelta(hours=1)
assert rt.date_from("+5mins", basedate) == basedate + timedelta(minutes=5)
assert rt.date_from("+20s", basedate) == basedate + timedelta(seconds=20)
assert rt.date_from("- 2 h", basedate) == basedate + timedelta(hours=-2)
assert rt.date_from("- 25 minutes", basedate) == basedate + timedelta(minutes=-25)
assert rt.date_from("-400 secs", basedate) == basedate + timedelta(seconds=-400)
assert rt.date_from("+1hr 5m", basedate) == basedate + timedelta(hours=1, minutes=5)
assert rt.date_from("-8hr 12m", basedate) == basedate + timedelta(hours=-8, minutes=-12)
assert rt.date_from("+1hr 5s", basedate) == basedate + timedelta(hours=1, seconds=5)
assert rt.date_from("+1hr 12m 5s", basedate) == basedate + timedelta(hours=1, minutes=12, seconds=5)
assert rt.date_from("-1hr 5s", basedate) == basedate + timedelta(hours=-1, seconds=-5)
assert rt.date_from("-1hr 12m 5s", basedate) == basedate + timedelta(hours=-1, minutes=-12, seconds=-5)
def test_relative_days():
# "next monday" on monday
assert relative_days(0, 0, 1) == 7
# "last monday" on monday
assert relative_days(0, 0, -1) == -7
# "next tuesday" on wednesday
assert relative_days(2, 1, 1) == 6
# "last tuesday" on wednesay
assert relative_days(2, 1, -1) == -1
# "last monday" on sunday
assert relative_days(6, 0, -1) == -6
# "next monday" on sunday
assert relative_days(6, 0, 1) == 1
# "next wednesday" on tuesday
assert relative_days(1, 2, 1) == 1
# "last wednesday" on tuesday
assert relative_days(1, 2, -1) == -6
# "last wednesday" on thursday
assert relative_days(3, 2, -1) == -1
# "next wednesday" on thursday
assert relative_days(3, 2, 1) == 6
# "last wednesday" on tuesday
assert relative_days(1, 2, -1) == -6
# "next wednesday" on tuesday
assert relative_days(1, 2, 1) == 1
def test_dayname(p=english.dayname):
assert_adatetime(p.date_from("next tuesday", basedate),
year=2010, month=9, day=21)
assert_adatetime(p.date_from("last tuesday", basedate),
year=2010, month=9, day=14)
assert_adatetime(p.date_from("next sunday", basedate),
year=2010, month=9, day=26)
assert_adatetime(p.date_from("last sun", basedate),
year=2010, month=9, day=19)
assert_adatetime(p.date_from("next th", basedate),
year=2010, month=9, day=23)
def test_reldate(p=english.plusdate):
assert p.date_from("+1y", basedate) == basedate + relativedelta(years=1)
assert p.date_from("+2mo", basedate) == basedate + relativedelta(months=2)
assert p.date_from("+3w", basedate) == basedate + relativedelta(weeks=3)
assert p.date_from("+5d", basedate) == basedate + relativedelta(days=5)
assert p.date_from("+5days", basedate) == basedate + relativedelta(days=5)
assert p.date_from("-6yr", basedate) == basedate + relativedelta(years=-6)
assert p.date_from("- 7 mons", basedate) == basedate + relativedelta(months=-7)
assert p.date_from("-8 wks", basedate) == basedate + relativedelta(weeks=-8)
assert p.date_from("- 9 dy", basedate) == basedate + relativedelta(days=-9)
assert p.date_from("+1y 12mo 400d", basedate) == basedate + relativedelta(years=1, months=12, days=400)
assert p.date_from("-7mo 8d", basedate) == basedate + relativedelta(months=-7, days=-8)
assert p.date_from("+5wks 2d", basedate) == basedate + relativedelta(weeks=5, days=2)
assert p.date_from("-1y 1w", basedate) == basedate + relativedelta(years=-1, weeks=-1)
assert p.date_from("+1y 2d 5h 12s", basedate) == basedate + relativedelta(years=1, days=2, hours=5, seconds=12)
def test_bundle_subs(p=english.bundle):
test_time(p)
test_dmy(p)
test_plustime(p)
test_dayname(p)
test_reldate(p)
def test_bundle(p=english.bundle):
assert_adatetime(p.date_from("mar 29 1972 2:45am", basedate),
year=1972, month=3, day=29, hour=2, minute=45)
assert_adatetime(p.date_from("16:10:45 14 February 2005", basedate),
year=2005, month=2, day=14, hour=16, minute=10, second=45)
assert_adatetime(p.date_from("1985 sept 12 12:01", basedate),
year=1985, month=9, day=12, hour=12, minute=1)
assert_adatetime(p.date_from("5pm 21st oct 2005", basedate),
year=2005, month=10, day=21, hour=17)
assert_adatetime(p.date_from("5:59:59pm next thur", basedate),
year=2010, month=9, day=23, hour=17, minute=59, second=59)
def test_ranges(p=english.torange):
assert_timespan(p.date_from("last tuesday to next tuesday", basedate),
dict(year=2010, month=9, day=14),
dict(year=2010, month=9, day=21))
assert_timespan(p.date_from("last monday to dec 25", basedate),
dict(year=2010, month=9, day=13),
dict(year=None, month=12, day=25))
assert_timespan(p.date_from("oct 25 to feb 14", basedate),
dict(year=None, month=10, day=25),
dict(year=None, month=2, day=14))
assert_timespan(p.date_from("3am oct 12 to 5pm", basedate),
dict(year=None, month=10, day=12, hour=3),
dict(year=None, month=None, day=None, hour=17))
assert_timespan(p.date_from("3am feb 12 to 5pm today", basedate),
dict(year=None, month=2, day=12, hour=3),
dict(year=2010, month=9, day=20, hour=17))
assert_timespan(p.date_from("feb to oct", basedate),
dict(year=None, month=2),
dict(year=None, month=10))
assert_timespan(p.date_from("oct 25 2005 11am to 5pm tomorrow", basedate),
dict(year=2005, month=10, day=25, hour=11),
dict(year=2010, month=9, day=21, hour=17))
assert_timespan(p.date_from("oct 5 2005 to november 20", basedate),
dict(year=2005, month=10, day=5),
dict(year=None, month=11, day=20))
assert_timespan(p.date_from("2007 to 2010", basedate),
dict(year=2007, month=None, day=None),
dict(year=2010, month=None, day=None))
assert_timespan(p.date_from("2007 to oct 12", basedate),
dict(year=2007, month=None, day=None),
dict(year=None, month=10, day=12))
assert_datespan(p.date_from("-2d to +1w", basedate),
basedate + relativedelta(days=-2),
basedate + relativedelta(weeks=1))
def test_all():
p = english.all
test_bundle_subs(p)
test_bundle(p)
test_ranges(p)
def test_final_dates(p=english):
assert_unamb(p.date_from("5:10pm", basedate),
year=2010, month=9, day=20, hour=17, minute=10)
assert p.date_from("may 32 2005", basedate) is None
assert p.date_from("2005 may 32", basedate) is None
assert p.date_from("2005-13-32", basedate) is None
def test_final_ranges(p=english):
assert_unamb_span(p.date_from("feb to nov", basedate),
dict(year=2010, month=2),
dict(year=2010, month=11))
# 2005 to 10 oct 2009 -> jan 1 2005 to oct 31 2009
assert_unamb_span(p.date_from("2005 to 10 oct 2009", basedate),
dict(year=2005),
dict(year=2009, month=10, day=10))
# jan 12 to oct 10 2009 -> jan 12 2009 to oct 10 2009
assert_unamb_span(p.date_from("jan 12 to oct 10 2009", basedate),
dict(year=2009, month=1, day=12),
dict(year=2009, month=10, day=10))
# jan to oct 2009 -> jan 1 2009 to oct 31 2009
assert_unamb_span(p.date_from("jan to oct 2009", basedate),
dict(year=2009, month=1),
dict(year=2009, month=10, day=31))
# mar 2005 to oct -> mar 1 2005 to oct 31 basedate.year
assert_unamb_span(p.date_from("mar 2005 to oct", basedate),
dict(year=2005, month=3),
dict(year=2010, month=10, day=31))
# jan 10 to jan 25 -> jan 10 basedate.year to jan 25 basedate.year
assert_unamb_span(p.date_from("jan 10 to jan 25", basedate),
dict(year=2010, month=1, day=10),
dict(year=2010, month=1, day=25))
# jan 2005 to feb 2009 -> jan 1 2005 to feb 28 2009
assert_unamb_span(p.date_from("jan 2005 to feb 2009", basedate),
dict(year=2005, month=1),
dict(year=2009, month=2))
# jan 5000 to mar -> jan 1 5000 to mar 5000
assert_unamb_span(p.date_from("jan 5000 to mar", basedate),
dict(year=5000, month=1),
dict(year=5000, month=3))
# jun 5000 to jan -> jun 1 5000 to jan 31 5001
assert_unamb_span(p.date_from("jun 5000 to jan", basedate),
dict(year=5000, month=6),
dict(year=5001, month=1))
# oct 2010 to feb -> oct 1 2010 to feb 28 2011
assert_unamb_span(p.date_from("oct 2010 to feb", basedate),
dict(year=2010, month=10),
dict(year=2011, month=2))
assert_unamb_span(p.date_from("5pm to 3am", basedate),
dict(year=2010, month=9, day=20, hour=17),
dict(year=2010, month=9, day=21, hour=3))
assert_unamb_span(p.date_from("5am to 3 am tomorrow", basedate),
dict(year=2010, month=9, day=20, hour=5),
dict(year=2010, month=9, day=21, hour=3))
assert_unamb_span(p.date_from("3am to 5 pm tomorrow", basedate),
dict(year=2010, month=9, day=21, hour=3),
dict(year=2010, month=9, day=21, hour=17))
assert_unamb_span(p.date_from("-2hrs to +20min", basedate),
dict(year=2010, month=9, day=20, hour=13, minute=16,
second=6, microsecond=454000),
dict(year=2010, month=9, day=20, hour=15, minute=36,
second=6, microsecond=454000))
# Swap
assert_unamb_span(p.date_from("oct 25 2009 to feb 14 2008", basedate),
dict(year=2008, month=2, day=14),
dict(year=2009, month=10, day=25))
assert_unamb_span(p.date_from("oct 25 5000 to tomorrow", basedate),
dict(year=2010, month=9, day=21),
dict(year=5000, month=10, day=25))
|
