From de66060b7aaa4e9a490df2979cea571815091688 Mon Sep 17 00:00:00 2001 From: Trenton H <797416+stumpylog@users.noreply.github.com> Date: Fri, 28 Jun 2024 10:26:10 -0700 Subject: [PATCH] Further improvements --- src/documents/tests/test_date_parsing.py | 38 +++++++++++++----------- 1 file changed, 20 insertions(+), 18 deletions(-) diff --git a/src/documents/tests/test_date_parsing.py b/src/documents/tests/test_date_parsing.py index 5bb29c8fa..f0afae543 100644 --- a/src/documents/tests/test_date_parsing.py +++ b/src/documents/tests/test_date_parsing.py @@ -361,24 +361,26 @@ class TestDate: For example 02.02.2018, 22 July 2022 and December 2021. But not 24-12-9999 because it's in the future...""" dates = list(parse_date_generator("", text)) - assert len(dates) == 3 - assert dates[0] == datetime.datetime(2018, 2, 2, 0, 0, tzinfo=settings_timezone) - assert dates[1] == datetime.datetime( - 2022, - 7, - 22, - 0, - 0, - tzinfo=settings_timezone, - ) - assert dates[2] == datetime.datetime( - 2021, - 12, - 1, - 0, - 0, - tzinfo=settings_timezone, - ) + + assert dates == [ + datetime.datetime(2018, 2, 2, 0, 0, tzinfo=settings_timezone), + datetime.datetime( + 2022, + 7, + 22, + 0, + 0, + tzinfo=settings_timezone, + ), + datetime.datetime( + 2021, + 12, + 1, + 0, + 0, + tzinfo=settings_timezone, + ), + ] def test_filename_date_parse_valid_ymd( self,