Skip to content

Commit 469dbe2

Browse files
committed
adapting tests for python 3.10
1 parent 2c764bd commit 469dbe2

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

tests/test_narwhalified.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,9 +180,14 @@ def _prepare_data(self):
180180
df_fractional_seconds1 = df_fractional_seconds_raw.clone()
181181
df_fractional_seconds1 = df_fractional_seconds1.to_native()
182182
if backend == "pandas":
183-
df_fractional_seconds1["date"] = pd.to_datetime(df_fractional_seconds1["date"])
183+
if is_pandas_version_newer_than_2:
184+
df_fractional_seconds1["date"] = pd.to_datetime(df_fractional_seconds1["date"])
185+
df_fractional_seconds1["dtime"] = pd.to_datetime(df_fractional_seconds1["dtime"])
186+
else:
187+
df_fractional_seconds1["date"] = pd.Series(np.array(df_fractional_seconds1["date"], dtype="datetime64[D]"))
188+
df_fractional_seconds1["dtime"] = pd.Series(np.array(df_fractional_seconds1["dtime"], dtype="datetime64[us]"))
189+
df_fractional_seconds1["dtime"] = df_fractional_seconds1["dtime"].dt.to_pydatetime()
184190
df_fractional_seconds1["date"] = df_fractional_seconds1["date"].apply(lambda x: x.date())
185-
df_fractional_seconds1["dtime"] = pd.to_datetime(df_fractional_seconds1["dtime"])
186191
df_fractional_seconds1["time"] = pd.to_datetime(df_fractional_seconds1["time"], format='%H:%M:%S.%f')
187192
df_fractional_seconds1["time"] = df_fractional_seconds1["time"].apply(lambda x: x.time())
188193
self.df_sas_fractional_seconds = df_fractional_seconds1

0 commit comments

Comments
 (0)