X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=tests%2Fdateparse_utils_test.py;h=6e539ff8ccce9592e09dd48a94b0d6b13bf0703f;hb=532df2c5b57c7517dfb3dddd8c1358fbadf8baf3;hp=3b3b80282932c01a18aa60f5ff7e8419aa35c223;hpb=b10d30a46e601c9ee1f843241f2d69a1f90f7a94;p=python_utils.git diff --git a/tests/dateparse_utils_test.py b/tests/dateparse_utils_test.py index 3b3b802..6e539ff 100755 --- a/tests/dateparse_utils_test.py +++ b/tests/dateparse_utils_test.py @@ -1,5 +1,9 @@ #!/usr/bin/env python3 +# © Copyright 2021-2022, Scott Gasch + +"""dateparse_utils unittest.""" + import datetime import random import re @@ -10,184 +14,135 @@ import pytz import dateparse.dateparse_utils as du import unittest_utils as uu - parsable_expressions = [ - ('today', - datetime.datetime(2021, 7, 2)), - ('tomorrow', - datetime.datetime(2021, 7, 3)), - ('yesterday', - datetime.datetime(2021, 7, 1)), - ('21:30', - datetime.datetime(2021, 7, 2, 21, 30, 0, 0)), - ('12:01am', - datetime.datetime(2021, 7, 2, 0, 1, 0, 0)), - ('12:02p', - datetime.datetime(2021, 7, 2, 12, 2, 0, 0)), - ('0:03', - datetime.datetime(2021, 7, 2, 0, 3, 0, 0)), - ('last wednesday', - datetime.datetime(2021, 6, 30)), - ('this wed', - datetime.datetime(2021, 7, 7)), - ('next wed', - datetime.datetime(2021, 7, 14)), - ('this coming tues', - datetime.datetime(2021, 7, 6)), - ('this past monday', - datetime.datetime(2021, 6, 28)), - ('4 days ago', - datetime.datetime(2021, 6, 28)), - ('4 mondays ago', - datetime.datetime(2021, 6, 7)), - ('4 months ago', - datetime.datetime(2021, 3, 2)), - ('3 days back', - datetime.datetime(2021, 6, 29)), - ('13 weeks from now', - datetime.datetime(2021, 10, 1)), - ('1 year from now', - datetime.datetime(2022, 7, 2)), - ('4 weeks from now', - datetime.datetime(2021, 7, 30)), - ('3 saturdays ago', - datetime.datetime(2021, 6, 12)), - ('4 months from today', - datetime.datetime(2021, 11, 2)), - ('4 years from yesterday', - datetime.datetime(2025, 7, 1)), - ('4 weeks from tomorrow', - datetime.datetime(2021, 7, 31)), - ('april 15, 2005', - datetime.datetime(2005, 4, 15)), - ('april 14', - datetime.datetime(2021, 4, 14)), - ('9:30am on last wednesday', - datetime.datetime(2021, 6, 30, 9, 30)), - ('2005/apr/15', - datetime.datetime(2005, 4, 15)), - ('2005 apr 15', - datetime.datetime(2005, 4, 15)), - ('the 1st wednesday in may', - datetime.datetime(2021, 5, 5)), - ('last sun of june', - datetime.datetime(2021, 6, 27)), - ('this Easter', - datetime.datetime(2021, 4, 4)), - ('last christmas', - datetime.datetime(2020, 12, 25)), - ('last Xmas', - datetime.datetime(2020, 12, 25)), - ('xmas, 1999', - datetime.datetime(1999, 12, 25)), - ('next mlk day', - datetime.datetime(2022, 1, 17)), - ('Halloween, 2020', - datetime.datetime(2020, 10, 31)), - ('5 work days after independence day', - datetime.datetime(2021, 7, 12)), - ('50 working days from last wed', - datetime.datetime(2021, 9, 10)), - ('25 working days before columbus day', - datetime.datetime(2021, 9, 3)), - ('today +1 week', - datetime.datetime(2021, 7, 9)), - ('sunday -3 weeks', - datetime.datetime(2021, 6, 13)), - ('4 weeks before xmas, 1999', - datetime.datetime(1999, 11, 27)), - ('3 days before new years eve, 2000', - datetime.datetime(2000, 12, 28)), - ('july 4th', - datetime.datetime(2021, 7, 4)), - ('the ides of march', - datetime.datetime(2021, 3, 15)), - ('the nones of april', - datetime.datetime(2021, 4, 5)), - ('the kalends of may', - datetime.datetime(2021, 5, 1)), - ('9/11/2001', - datetime.datetime(2001, 9, 11)), - ('4 sundays before veterans\' day', - datetime.datetime(2021, 10, 17)), - ('xmas eve', - datetime.datetime(2021, 12, 24)), - ('this friday at 5pm', - datetime.datetime(2021, 7, 9, 17, 0, 0)), - ('presidents day', - datetime.datetime(2021, 2, 15)), - ('memorial day, 1921', - datetime.datetime(1921, 5, 30)), - ('today -4 wednesdays', - datetime.datetime(2021, 6, 9)), - ('thanksgiving', - datetime.datetime(2021, 11, 25)), - ('2 sun in jun', - datetime.datetime(2021, 6, 13)), - ('easter -40 days', - datetime.datetime(2021, 2, 23)), - ('easter +39 days', - datetime.datetime(2021, 5, 13)), - ('1st tuesday in nov, 2024', - datetime.datetime(2024, 11, 5)), - ('2 days before last xmas at 3:14:15.92a', - datetime.datetime(2020, 12, 23, 3, 14, 15, 92)), - ('3 weeks after xmas, 1995 at midday', - datetime.datetime(1996, 1, 15, 12, 0, 0)), - ('4 months before easter, 1992 at midnight', - datetime.datetime(1991, 12, 19)), - ('5 months before halloween, 1995 at noon', - datetime.datetime(1995, 5, 31, 12)), - ('4 days before last wednesday', - datetime.datetime(2021, 6, 26)), - ('44 months after today', - datetime.datetime(2025, 3, 2)), - ('44 years before today', - datetime.datetime(1977, 7, 2)), - ('44 weeks ago', - datetime.datetime(2020, 8, 28)), - ('15 minutes to 3am', - datetime.datetime(2021, 7, 2, 2, 45)), - ('quarter past 4pm', - datetime.datetime(2021, 7, 2, 16, 15)), - ('half past 9', - datetime.datetime(2021, 7, 2, 9, 30)), - ('4 seconds to midnight', - datetime.datetime(2021, 7, 1, 23, 59, 56)), - ('4 seconds to midnight, tomorrow', - datetime.datetime(2021, 7, 2, 23, 59, 56)), - ('2021/apr/15T21:30:44.55', - datetime.datetime(2021, 4, 15, 21, 30, 44, 55)), - ('2021/apr/15 at 21:30:44.55', - datetime.datetime(2021, 4, 15, 21, 30, 44, 55)), - ('2021/4/15 at 21:30:44.55', - datetime.datetime(2021, 4, 15, 21, 30, 44, 55)), - ('2021/04/15 at 21:30:44.55', - datetime.datetime(2021, 4, 15, 21, 30, 44, 55)), - ('2021/04/15 at 21:30:44.55Z', - datetime.datetime(2021, 4, 15, 21, 30, 44, 55, - tzinfo=pytz.timezone('UTC'))), - ('2021/04/15 at 21:30:44.55EST', - datetime.datetime(2021, 4, 15, 21, 30, 44, 55, - tzinfo=pytz.timezone('EST'))), - ('13 days after last memorial day at 12 seconds before 4pm', - datetime.datetime(2020, 6, 7, 15, 59, 48)), - (' 2 days before yesterday at 9am ', - datetime.datetime(2021, 6, 29, 9)), - ('-3 days before today', - datetime.datetime(2021, 7, 5)), - ('3 days before yesterday at midnight EST', - datetime.datetime(2021, 6, 28, tzinfo=pytz.timezone('EST'))) + ('today', datetime.datetime(2021, 7, 2)), + ('tomorrow', datetime.datetime(2021, 7, 3)), + ('yesterday', datetime.datetime(2021, 7, 1)), + ('21:30', datetime.datetime(2021, 7, 2, 21, 30, 0, 0)), + ('12:01am', datetime.datetime(2021, 7, 2, 0, 1, 0, 0)), + ('12:02p', datetime.datetime(2021, 7, 2, 12, 2, 0, 0)), + ('0:03', datetime.datetime(2021, 7, 2, 0, 3, 0, 0)), + ('last wednesday', datetime.datetime(2021, 6, 30)), + ('this wed', datetime.datetime(2021, 7, 7)), + ('next wed', datetime.datetime(2021, 7, 14)), + ('this coming tues', datetime.datetime(2021, 7, 6)), + ('this past monday', datetime.datetime(2021, 6, 28)), + ('4 days ago', datetime.datetime(2021, 6, 28)), + ('4 mondays ago', datetime.datetime(2021, 6, 7)), + ('4 months ago', datetime.datetime(2021, 3, 2)), + ('3 days back', datetime.datetime(2021, 6, 29)), + ('13 weeks from now', datetime.datetime(2021, 10, 1)), + ('1 year from now', datetime.datetime(2022, 7, 2)), + ('4 weeks from now', datetime.datetime(2021, 7, 30)), + ('3 saturdays ago', datetime.datetime(2021, 6, 12)), + ('4 months from today', datetime.datetime(2021, 11, 2)), + ('4 years from yesterday', datetime.datetime(2025, 7, 1)), + ('4 weeks from tomorrow', datetime.datetime(2021, 7, 31)), + ('april 15, 2005', datetime.datetime(2005, 4, 15)), + ('april 14', datetime.datetime(2021, 4, 14)), + ('9:30am on last wednesday', datetime.datetime(2021, 6, 30, 9, 30)), + ('2005/apr/15', datetime.datetime(2005, 4, 15)), + ('2005 apr 15', datetime.datetime(2005, 4, 15)), + ('the 1st wednesday in may', datetime.datetime(2021, 5, 5)), + ('last sun of june', datetime.datetime(2021, 6, 27)), + ('this Easter', datetime.datetime(2021, 4, 4)), + ('last christmas', datetime.datetime(2020, 12, 25)), + ('last Xmas', datetime.datetime(2020, 12, 25)), + ('xmas, 1999', datetime.datetime(1999, 12, 25)), + ('next mlk day', datetime.datetime(2022, 1, 17)), + ('Halloween, 2020', datetime.datetime(2020, 10, 31)), + ('5 work days after independence day', datetime.datetime(2021, 7, 12)), + ('50 working days from last wed', datetime.datetime(2021, 9, 10)), + ('25 working days before columbus day', datetime.datetime(2021, 9, 3)), + ('today +1 week', datetime.datetime(2021, 7, 9)), + ('sunday -3 weeks', datetime.datetime(2021, 6, 13)), + ('4 weeks before xmas, 1999', datetime.datetime(1999, 11, 27)), + ('3 days before new years eve, 2000', datetime.datetime(2000, 12, 28)), + ('july 4th', datetime.datetime(2021, 7, 4)), + ('the ides of march', datetime.datetime(2021, 3, 15)), + ('the nones of april', datetime.datetime(2021, 4, 5)), + ('the kalends of may', datetime.datetime(2021, 5, 1)), + ('9/11/2001', datetime.datetime(2001, 9, 11)), + ('4 sundays before veterans\' day', datetime.datetime(2021, 10, 17)), + ('xmas eve', datetime.datetime(2021, 12, 24)), + ('this friday at 5pm', datetime.datetime(2021, 7, 9, 17, 0, 0)), + ('presidents day', datetime.datetime(2021, 2, 15)), + ('memorial day, 1921', datetime.datetime(1921, 5, 30)), + ('today -4 wednesdays', datetime.datetime(2021, 6, 9)), + ('thanksgiving', datetime.datetime(2021, 11, 25)), + ('2 sun in jun', datetime.datetime(2021, 6, 13)), + ('easter -40 days', datetime.datetime(2021, 2, 23)), + ('easter +39 days', datetime.datetime(2021, 5, 13)), + ('1st tuesday in nov, 2024', datetime.datetime(2024, 11, 5)), + ( + '2 days before last xmas at 3:14:15.92a', + datetime.datetime(2020, 12, 23, 3, 14, 15, 92), + ), + ( + '3 weeks after xmas, 1995 at midday', + datetime.datetime(1996, 1, 15, 12, 0, 0), + ), + ( + '4 months before easter, 1992 at midnight', + datetime.datetime(1991, 12, 19), + ), + ( + '5 months before halloween, 1995 at noon', + datetime.datetime(1995, 5, 31, 12), + ), + ('4 days before last wednesday', datetime.datetime(2021, 6, 26)), + ('44 months after today', datetime.datetime(2025, 3, 2)), + ('44 years before today', datetime.datetime(1977, 7, 2)), + ('44 weeks ago', datetime.datetime(2020, 8, 28)), + ('15 minutes to 3am', datetime.datetime(2021, 7, 2, 2, 45)), + ('quarter past 4pm', datetime.datetime(2021, 7, 2, 16, 15)), + ('half past 9', datetime.datetime(2021, 7, 2, 9, 30)), + ('4 seconds to midnight', datetime.datetime(2021, 7, 1, 23, 59, 56)), + ( + '4 seconds to midnight, tomorrow', + datetime.datetime(2021, 7, 2, 23, 59, 56), + ), + ('2021/apr/15T21:30:44.55', datetime.datetime(2021, 4, 15, 21, 30, 44, 55)), + ( + '2021/apr/15 at 21:30:44.55', + datetime.datetime(2021, 4, 15, 21, 30, 44, 55), + ), + ( + '2021/4/15 at 21:30:44.55', + datetime.datetime(2021, 4, 15, 21, 30, 44, 55), + ), + ( + '2021/04/15 at 21:30:44.55', + datetime.datetime(2021, 4, 15, 21, 30, 44, 55), + ), + ( + '2021/04/15 at 21:30:44.55Z', + datetime.datetime(2021, 4, 15, 21, 30, 44, 55, tzinfo=pytz.timezone('UTC')), + ), + ( + '2021/04/15 at 21:30:44.55EST', + datetime.datetime(2021, 4, 15, 21, 30, 44, 55, tzinfo=pytz.timezone('EST')), + ), + ( + '13 days after last memorial day at 12 seconds before 4pm', + datetime.datetime(2020, 6, 7, 15, 59, 48), + ), + ( + ' 2 days before yesterday at 9am ', + datetime.datetime(2021, 6, 29, 9), + ), + ('-3 days before today', datetime.datetime(2021, 7, 5)), + ( + '3 days before yesterday at midnight EST', + datetime.datetime(2021, 6, 28, tzinfo=pytz.timezone('EST')), + ), ] class TestDateparseUtils(unittest.TestCase): - @uu.check_method_for_perf_regressions def test_dateparsing(self): - dp = du.DateParser( - override_now_for_test_purposes = datetime.datetime(2021, 7, 2) - ) + dp = du.DateParser(override_now_for_test_purposes=datetime.datetime(2021, 7, 2)) for (txt, expected_dt) in parsable_expressions: try: @@ -196,28 +151,26 @@ class TestDateparseUtils(unittest.TestCase): self.assertEqual( actual_dt, expected_dt, - f'"{txt}", got "{actual_dt}" while expecting "{expected_dt}"' + f'"{txt}", got "{actual_dt}" while expecting "{expected_dt}"', ) except du.ParseException: self.fail(f'Expected "{txt}" to parse successfully.') def test_whitespace_handling(self): - dp = du.DateParser( - override_now_for_test_purposes = datetime.datetime(2021, 7, 2) - ) + dp = du.DateParser(override_now_for_test_purposes=datetime.datetime(2021, 7, 2)) for (txt, expected_dt) in parsable_expressions: try: txt = f' {txt} ' i = random.randint(2, 5) replacement = ' ' * i - txt = re.sub('\s', replacement, txt) + txt = re.sub(r'\s', replacement, txt) actual_dt = dp.parse(txt) self.assertIsNotNone(actual_dt) self.assertEqual( actual_dt, expected_dt, - f'"{txt}", got "{actual_dt}" while expecting "{expected_dt}"' + f'"{txt}", got "{actual_dt}" while expecting "{expected_dt}"', ) except du.ParseException: self.fail(f'Expected "{txt}" to parse successfully.')