add datetime field

bmsuisse · Mar 13, 2024 · dbbb3e8 · dbbb3e8
1 parent 91ff255
commit dbbb3e8
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_polars_read.py b/tests/test_polars_read.py
@@ -2,6 +2,7 @@
 import pytest
 from polars.testing import assert_frame_equal
 from deltalake2db import polars_scan_delta
+from datetime import datetime
 
 
 @pytest.fixture()
@@ -21,6 +22,17 @@ def data_batch_1():
                 201008,
                 201009,
             ],
+            "datetime": [
+                datetime.fromisoformat("2010-01-01"),
+                datetime.fromisoformat("2010-02-01"),
+                datetime.fromisoformat("2010-03-01"),
+                datetime.fromisoformat("2010-04-01"),
+                datetime.fromisoformat("2010-05-01"),
+                datetime.fromisoformat("2010-06-01"),
+                datetime.fromisoformat("2010-07-01"),
+                datetime.fromisoformat("2010-08-01"),
+                datetime.fromisoformat("2010-09-01"),
+            ],
             "static_part": ["A", "A", "A", "B", "B", "B", "C", "C", "C"],
         }
     )