Skip to content

Commit fe4e174

Browse files
committed
remove importorskip
1 parent 000a159 commit fe4e174

File tree

1 file changed

+12
-17
lines changed

1 file changed

+12
-17
lines changed

tools/pythonpkg/tests/fast/pandas/test_datetime_time.py

Lines changed: 12 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,14 @@
11
import pandas as pd
22
import duckdb
3-
import datetime
43
import numpy as np
54
import pytest
6-
5+
from datetime import datetime, timezone, time, timedelta
76

87
class TestDateTimeTime(object):
98

109
def test_time_high(self, duckdb_cursor):
1110
duckdb_time = duckdb.query("SELECT make_time(23, 1, 34.234345) AS '0'").df()
12-
data = [datetime.time(hour=23, minute=1, second=34, microsecond=234345)]
11+
data = [time(hour=23, minute=1, second=34, microsecond=234345)]
1312
df_in = pd.DataFrame(
1413
{'0': pd.Series(data=data, dtype='object')}
1514
)
@@ -18,7 +17,7 @@ def test_time_high(self, duckdb_cursor):
1817

1918
def test_time_low(self, duckdb_cursor):
2019
duckdb_time = duckdb.query("SELECT make_time(00, 01, 1.000) AS '0'").df()
21-
data = [datetime.time(hour=0, minute=1, second=1)]
20+
data = [time(hour=0, minute=1, second=1)]
2221
df_in = pd.DataFrame(
2322
{'0': pd.Series(data=data, dtype='object')}
2423
)
@@ -28,9 +27,9 @@ def test_time_low(self, duckdb_cursor):
2827
def test_time_timezone_regular(self, duckdb_cursor):
2928
duckdb_time = duckdb.query("SELECT make_time(00, 01, 1.000) AS '0'").df()
3029
# time is 3 hours ahead of UTC
31-
offset = datetime.timedelta(hours=3)
32-
timezone = datetime.timezone(offset)
33-
data = [datetime.time(hour=3, minute=1, second=1, tzinfo=timezone)]
30+
offset = timedelta(hours=3)
31+
tz = timezone(offset)
32+
data = [time(hour=3, minute=1, second=1, tzinfo=tz)]
3433
df_in = pd.DataFrame(
3534
{'0': pd.Series(data=data, dtype='object')}
3635
)
@@ -40,9 +39,9 @@ def test_time_timezone_regular(self, duckdb_cursor):
4039
def test_time_timezone_negative_extreme(self, duckdb_cursor):
4140
duckdb_time = duckdb.query("SELECT make_time(12, 01, 1.000) AS '0'").df()
4241
# time is 14 hours behind UTC
43-
offset = datetime.timedelta(hours=-14)
44-
timezone = datetime.timezone(offset)
45-
data = [datetime.time(hour=22, minute=1, second=1, tzinfo=timezone)]
42+
offset = timedelta(hours=-14)
43+
tz = timezone(offset)
44+
data = [time(hour=22, minute=1, second=1, tzinfo=tz)]
4645
df_in = pd.DataFrame(
4746
{'0': pd.Series(data=data, dtype='object')}
4847
)
@@ -52,9 +51,9 @@ def test_time_timezone_negative_extreme(self, duckdb_cursor):
5251
def test_time_timezone_positive_extreme(self, duckdb_cursor):
5352
duckdb_time = duckdb.query("SELECT make_time(12, 01, 1.000) AS '0'").df()
5453
# time is 20 hours ahead of UTC
55-
offset = datetime.timedelta(hours=20)
56-
timezone = datetime.timezone(offset)
57-
data = [datetime.time(hour=8, minute=1, second=1, tzinfo=timezone)]
54+
offset = timedelta(hours=20)
55+
tz = timezone(offset)
56+
data = [time(hour=8, minute=1, second=1, tzinfo=tz)]
5857
df_in = pd.DataFrame(
5958
{'0': pd.Series(data=data, dtype='object')}
6059
)
@@ -71,10 +70,6 @@ def test_pandas_datetime_overflow(self):
7170
res = duckdb_con.execute("select * from test").df()
7271

7372
def test_timezone_datetime(self):
74-
module = pytest.importorskip("datetime")
75-
timezone = module.timezone
76-
datetime = module.datetime
77-
7873
con = duckdb.connect()
7974

8075
dt = datetime.now(timezone.utc).replace(microsecond=0)

0 commit comments

Comments
 (0)