@@ -16,7 +16,7 @@ html-table-processing: none
1616import pointblank as pb
1717import polars as pl
1818from datetime import date, datetime
19- import pytz
19+ from zoneinfo import ZoneInfo
2020
2121# Create sample data with various temporal data types
2222temporal_data = pl.DataFrame({
@@ -33,10 +33,10 @@ temporal_data = pl.DataFrame({
3333 datetime(2024, 3, 20, 17, 22, 45)
3434 ],
3535 "event_time_tz": [
36- datetime(2023, 1, 15, 9, 0, tzinfo=pytz.timezone ("America/New_York")),
37- datetime(2023, 6, 10, 12, 30, tzinfo=pytz.timezone ("America/New_York")),
38- datetime(2023, 12, 5, 15, 45, tzinfo=pytz.timezone ("America/New_York")),
39- datetime(2024, 3, 20, 18, 15, tzinfo=pytz.timezone ("America/New_York"))
36+ datetime(2023, 1, 15, 9, 0, tzinfo=ZoneInfo ("America/New_York")),
37+ datetime(2023, 6, 10, 12, 30, tzinfo=ZoneInfo ("America/New_York")),
38+ datetime(2023, 12, 5, 15, 45, tzinfo=ZoneInfo ("America/New_York")),
39+ datetime(2024, 3, 20, 18, 15, tzinfo=ZoneInfo ("America/New_York"))
4040 ],
4141 "order_id": [1001, 1002, 1003, 1004],
4242 "amount": [150.0, 275.5, 89.99, 420.00]
@@ -57,7 +57,7 @@ validation = (
5757 )
5858 .col_vals_ge(
5959 columns="event_time_tz",
60- value=datetime(2023, 1, 1, 8, 0, tzinfo=pytz.timezone ("America/New_York")),
60+ value=datetime(2023, 1, 1, 8, 0, tzinfo=ZoneInfo ("America/New_York")),
6161 brief="Timezone-aware events after 8 AM Eastern"
6262 )
6363 .col_schema_match(
@@ -82,7 +82,7 @@ validation
8282import pointblank as pb
8383import polars as pl
8484from datetime import date, datetime
85- import pytz
85+ from zoneinfo import ZoneInfo
8686
8787# Create sample data with various temporal data types
8888temporal_data = pl.DataFrame({
@@ -99,10 +99,10 @@ temporal_data = pl.DataFrame({
9999 datetime(2024 , 3 , 20 , 17 , 22 , 45 )
100100 ],
101101 " event_time_tz" : [
102- datetime(2023 , 1 , 15 , 9 , 0 , tzinfo = pytz.timezone (" America/New_York" )),
103- datetime(2023 , 6 , 10 , 12 , 30 , tzinfo = pytz.timezone (" America/New_York" )),
104- datetime(2023 , 12 , 5 , 15 , 45 , tzinfo = pytz.timezone (" America/New_York" )),
105- datetime(2024 , 3 , 20 , 18 , 15 , tzinfo = pytz.timezone (" America/New_York" ))
102+ datetime(2023 , 1 , 15 , 9 , 0 , tzinfo = ZoneInfo (" America/New_York" )),
103+ datetime(2023 , 6 , 10 , 12 , 30 , tzinfo = ZoneInfo (" America/New_York" )),
104+ datetime(2023 , 12 , 5 , 15 , 45 , tzinfo = ZoneInfo (" America/New_York" )),
105+ datetime(2024 , 3 , 20 , 18 , 15 , tzinfo = ZoneInfo (" America/New_York" ))
106106 ],
107107 " order_id" : [1001 , 1002 , 1003 , 1004 ],
108108 " amount" : [150.0 , 275.5 , 89.99 , 420.00 ]
@@ -123,7 +123,7 @@ validation = (
123123 )
124124 .col_vals_ge(
125125 columns = " event_time_tz" ,
126- value = datetime(2023 , 1 , 1 , 8 , 0 , tzinfo = pytz.timezone (" America/New_York" )),
126+ value = datetime(2023 , 1 , 1 , 8 , 0 , tzinfo = ZoneInfo (" America/New_York" )),
127127 brief = " Timezone-aware events after 8 AM Eastern"
128128 )
129129 .col_schema_match(
0 commit comments