Skip to content

Commit 1fbd5ad

Browse files
committed
Fix formatting
1 parent 9e168b5 commit 1fbd5ad

1 file changed

Lines changed: 32 additions & 20 deletions

File tree

test/test_scan_pandas.py

Lines changed: 32 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -776,20 +776,40 @@ def test_scan_pandas_datetime_nat(conn_db_empty: ConnDB) -> None:
776776
df = pd.DataFrame(
777777
{
778778
"dt_s": np.array(
779-
[valid_ts, np.datetime64("NaT", "s"), np.datetime64("NaT", "s"),
780-
valid_ts2], dtype="datetime64[s]"
779+
[
780+
valid_ts,
781+
np.datetime64("NaT", "s"),
782+
np.datetime64("NaT", "s"),
783+
valid_ts2,
784+
],
785+
dtype="datetime64[s]",
781786
),
782787
"dt_ms": np.array(
783-
[valid_ts, np.datetime64("NaT", "ms"), np.datetime64("NaT", "ms"),
784-
valid_ts2], dtype="datetime64[ms]"
788+
[
789+
valid_ts,
790+
np.datetime64("NaT", "ms"),
791+
np.datetime64("NaT", "ms"),
792+
valid_ts2,
793+
],
794+
dtype="datetime64[ms]",
785795
),
786796
"dt_us": np.array(
787-
[valid_ts, np.datetime64("NaT", "us"), np.datetime64("NaT", "us"),
788-
valid_ts2], dtype="datetime64[us]"
797+
[
798+
valid_ts,
799+
np.datetime64("NaT", "us"),
800+
np.datetime64("NaT", "us"),
801+
valid_ts2,
802+
],
803+
dtype="datetime64[us]",
789804
),
790805
"dt_ns": np.array(
791-
[valid_ts, np.datetime64("NaT", "ns"), np.datetime64("NaT", "ns"),
792-
valid_ts2], dtype="datetime64[ns]"
806+
[
807+
valid_ts,
808+
np.datetime64("NaT", "ns"),
809+
np.datetime64("NaT", "ns"),
810+
valid_ts2,
811+
],
812+
dtype="datetime64[ns]",
793813
),
794814
}
795815
)
@@ -818,9 +838,7 @@ def test_scan_pandas_timedelta_nat(conn_db_empty: ConnDB) -> None:
818838
valid_td2 = np.timedelta64(5000000000, "ns") # 5 seconds
819839
df = pd.DataFrame(
820840
{
821-
"td": np.array(
822-
[valid_td, nat, nat, valid_td2], dtype="timedelta64[ns]"
823-
),
841+
"td": np.array([valid_td, nat, nat, valid_td2], dtype="timedelta64[ns]"),
824842
}
825843
)
826844
res = conn.execute("LOAD FROM df RETURN *")
@@ -838,9 +856,7 @@ def test_scan_pandas_timedelta_nat(conn_db_empty: ConnDB) -> None:
838856
def test_copy_from_datetime_nat(conn_db_empty: ConnDB) -> None:
839857
"""Test that COPY FROM with NaT datetime stores NULL in the table."""
840858
conn, _ = conn_db_empty
841-
conn.execute(
842-
"CREATE NODE TABLE Test (id INT64, ts TIMESTAMP, PRIMARY KEY (id))"
843-
)
859+
conn.execute("CREATE NODE TABLE Test (id INT64, ts TIMESTAMP, PRIMARY KEY (id))")
844860
valid_ts = np.datetime64("2024-01-15T10:30:00")
845861
nat = np.datetime64("NaT", "ns")
846862
df = pd.DataFrame(
@@ -868,9 +884,7 @@ def test_copy_from_datetime_nat(conn_db_empty: ConnDB) -> None:
868884
def test_copy_from_timedelta_nat(conn_db_empty: ConnDB) -> None:
869885
"""Test that COPY FROM with NaT timedelta stores NULL in the table."""
870886
conn, _ = conn_db_empty
871-
conn.execute(
872-
"CREATE NODE TABLE Test (id INT64, dur INTERVAL, PRIMARY KEY (id))"
873-
)
887+
conn.execute("CREATE NODE TABLE Test (id INT64, dur INTERVAL, PRIMARY KEY (id))")
874888
nat = np.timedelta64("NaT", "ns")
875889
valid_td = np.timedelta64(3600000000000, "ns") # 1 hour
876890
df = pd.DataFrame(
@@ -900,9 +914,7 @@ def test_copy_from_datetime_none(conn_db_empty: ConnDB) -> None:
900914
Pandas auto-infers the column as datetime64[ns] and converts None to NaT
901915
"""
902916
conn, _ = conn_db_empty
903-
conn.execute(
904-
"CREATE NODE TABLE Test (id INT64, ts TIMESTAMP, PRIMARY KEY (id))"
905-
)
917+
conn.execute("CREATE NODE TABLE Test (id INT64, ts TIMESTAMP, PRIMARY KEY (id))")
906918
df = pd.DataFrame(
907919
{
908920
"id": [1, 2],

0 commit comments

Comments
 (0)