From 8ead01a9237d6bc56db540ba08766a5158b0b94e Mon Sep 17 00:00:00 2001 From: Allen Porter Date: Wed, 11 Sep 2024 21:13:10 -0700 Subject: [PATCH] Add test coverage for computed duration (#435) --- tests/test_journal.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/test_journal.py b/tests/test_journal.py index d7fabc5..07177ae 100644 --- a/tests/test_journal.py +++ b/tests/test_journal.py @@ -50,3 +50,19 @@ def test_start_datetime() -> None: assert ts assert ts.start == datetime.datetime(2022, 8, 7, 0, 0, 0, tzinfo=datetime.timezone.utc) assert ts.end == datetime.datetime(2022, 8, 8, 0, 0, 0, tzinfo=datetime.timezone.utc) + + +def test_computed_duration_date() -> None: + """Test computed duration for a date.""" + + journal = Journal(start=datetime.date(2022, 8, 7,)) + assert journal.start + assert journal.computed_duration == datetime.timedelta(days=1) + + +def test_computed_duration_datetime() -> None: + """Test computed duration for a datetime.""" + + journal = Journal(start=datetime.datetime(2022, 8, 7, 0, 0, 0)) + assert journal.start + assert journal.computed_duration == datetime.timedelta(hours=1)