@@ -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:
838856def 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:
868884def 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