22
33from __future__ import annotations
44
5- from datetime import datetime , timedelta , timezone
5+ from datetime import UTC , datetime , timedelta
66
77import pytest
88
@@ -72,39 +72,39 @@ class TestParseTimeSpec:
7272 def test_today_keyword (self ) -> None :
7373 """Test 'today' keyword returns start of today in UTC."""
7474 result = parse_time_spec ("today" )
75- now = datetime .now (tz = timezone . utc )
75+ now = datetime .now (tz = UTC )
7676 expected = now .replace (hour = 0 , minute = 0 , second = 0 , microsecond = 0 )
7777 assert result == expected
7878
7979 def test_yesterday_keyword (self ) -> None :
8080 """Test 'yesterday' keyword returns start of yesterday in UTC."""
8181 result = parse_time_spec ("yesterday" )
82- now = datetime .now (tz = timezone . utc )
82+ now = datetime .now (tz = UTC )
8383 expected = (now - timedelta (days = 1 )).replace (hour = 0 , minute = 0 , second = 0 , microsecond = 0 )
8484 assert result == expected
8585
8686 def test_now_keyword (self ) -> None :
8787 """Test 'now' keyword returns current time in UTC."""
88- before = datetime .now (tz = timezone . utc )
88+ before = datetime .now (tz = UTC )
8989 result = parse_time_spec ("now" )
90- after = datetime .now (tz = timezone . utc )
90+ after = datetime .now (tz = UTC )
9191 assert before <= result <= after
9292
9393 def test_relative_times (self ) -> None :
9494 """Test relative time specifications."""
95- before = datetime .now (tz = timezone . utc )
95+ before = datetime .now (tz = UTC )
9696 result = parse_time_spec ("7d" )
97- after = datetime .now (tz = timezone . utc )
97+ after = datetime .now (tz = UTC )
9898
9999 expected_min = before - timedelta (days = 7 )
100100 expected_max = after - timedelta (days = 7 )
101101 assert expected_min <= result <= expected_max
102102
103103 def test_relative_hours (self ) -> None :
104104 """Test relative hour specifications."""
105- before = datetime .now (tz = timezone . utc )
105+ before = datetime .now (tz = UTC )
106106 result = parse_time_spec ("2h" )
107- after = datetime .now (tz = timezone . utc )
107+ after = datetime .now (tz = UTC )
108108
109109 expected_min = before - timedelta (hours = 2 )
110110 expected_max = after - timedelta (hours = 2 )
@@ -113,19 +113,19 @@ def test_relative_hours(self) -> None:
113113 def test_iso_date (self ) -> None :
114114 """Test ISO date parsing."""
115115 result = parse_time_spec ("2024-01-15" )
116- expected = datetime (2024 , 1 , 15 , 0 , 0 , 0 , tzinfo = timezone . utc )
116+ expected = datetime (2024 , 1 , 15 , 0 , 0 , 0 , tzinfo = UTC )
117117 assert result == expected
118118
119119 def test_iso_datetime_with_t (self ) -> None :
120120 """Test ISO datetime parsing with T separator."""
121121 result = parse_time_spec ("2024-01-15T10:30:00" )
122- expected = datetime (2024 , 1 , 15 , 10 , 30 , 0 , tzinfo = timezone . utc )
122+ expected = datetime (2024 , 1 , 15 , 10 , 30 , 0 , tzinfo = UTC )
123123 assert result == expected
124124
125125 def test_iso_datetime_with_space (self ) -> None :
126126 """Test ISO datetime parsing with space separator."""
127127 result = parse_time_spec ("2024-01-15 10:30:00" )
128- expected = datetime (2024 , 1 , 15 , 10 , 30 , 0 , tzinfo = timezone . utc )
128+ expected = datetime (2024 , 1 , 15 , 10 , 30 , 0 , tzinfo = UTC )
129129 assert result == expected
130130
131131 def test_case_insensitive_keywords (self ) -> None :
@@ -154,27 +154,27 @@ class TestParseDateSpec:
154154 def test_today_keyword (self ) -> None :
155155 """Test 'today' keyword returns today's date."""
156156 result = parse_date_spec ("today" )
157- now = datetime .now (tz = timezone . utc )
157+ now = datetime .now (tz = UTC )
158158 assert result == now .strftime ("%Y-%m-%d" )
159159
160160 def test_yesterday_keyword (self ) -> None :
161161 """Test 'yesterday' keyword returns yesterday's date."""
162162 result = parse_date_spec ("yesterday" )
163- now = datetime .now (tz = timezone . utc )
163+ now = datetime .now (tz = UTC )
164164 expected = (now - timedelta (days = 1 )).strftime ("%Y-%m-%d" )
165165 assert result == expected
166166
167167 def test_relative_days (self ) -> None :
168168 """Test relative day specifications."""
169169 result = parse_date_spec ("7d" )
170- now = datetime .now (tz = timezone . utc )
170+ now = datetime .now (tz = UTC )
171171 expected = (now - timedelta (days = 7 )).strftime ("%Y-%m-%d" )
172172 assert result == expected
173173
174174 def test_relative_30_days (self ) -> None :
175175 """Test 30 days ago."""
176176 result = parse_date_spec ("30d" )
177- now = datetime .now (tz = timezone . utc )
177+ now = datetime .now (tz = UTC )
178178 expected = (now - timedelta (days = 30 )).strftime ("%Y-%m-%d" )
179179 assert result == expected
180180
0 commit comments