Add date/datetime tests for a DuckDB table

Author: rich-iannone

data_raw/x-02-duckdb.qmd

@@ -9,16 +9,30 @@ import polars as pl
 import duckdb
 
 tbl_xyz = pl.DataFrame({"x": [1, 2, 3, 4], "y": [4, 5, 6, 7], "z": [8, 8, 8, 8]})
-tbl_xyz_missing = pl.DataFrame(
-    {"x": [1, 2, None, 4], "y": [4, None, 6, 7], "z": [8, None, 8, 8]}
-)
+tbl_xyz_missing = pl.DataFrame({"x": [1, 2, None, 4], "y": [4, None, 6, 7], "z": [8, None, 8, 8]})
 tbl_dates_times_text = pl.DataFrame(
     {
         "date": ["2021-01-01", "2021-02-01", None],
         "dttm": ["2021-01-01 00:00:00", None, "2021-02-01 00:00:00"],
         "text": [None, "5-egh-163", "8-kdg-938"],
     }
 )
+tbl_true_dates_times = pl.DataFrame(
+    {
+        "date_1": ["2021-01-01", "2021-02-01"],
+        "date_2": ["2021-02-01", "2021-03-01"],
+        "dttm_1": ["2021-01-01 02:30:00", "2021-02-01 02:30:00"],
+        "dttm_2": ["2021-02-01 03:30:00", "2021-03-01 03:30:00"],
+    }
+).with_columns(
+    [
+        pl.col("date_1").str.to_date(),
+        pl.col("date_2").str.to_date(),
+        pl.col("dttm_1").str.to_datetime(),
+        pl.col("dttm_2").str.to_datetime(),
+    ]
+)
+
 small_table = pb.load_dataset(dataset="small_table", tbl_type="polars")
 game_revenue = pb.load_dataset(dataset="game_revenue", tbl_type="polars")
 nycflights = pb.load_dataset(dataset="nycflights", tbl_type="polars")
@@ -47,6 +61,13 @@ with duckdb.connect(database="tbl_dates_times_text.ddb", read_only=False) as con
     """)
 ```
 
+```{python}
+with duckdb.connect(database="tbl_true_dates_times.ddb", read_only=False) as con:
+    con.execute(f"""
+        CREATE TABLE IF NOT EXISTS 'tbl_true_dates_times' AS SELECT * FROM tbl_true_dates_times;
+    """)
+```
+
 ```{python}
 with duckdb.connect(database="small_table.ddb", read_only=False) as con:
     con.execute(f"""

tests/tbl_files/tbl_true_dates_times.ddb

@@ -85,6 +85,7 @@
 TBL_TRUE_DATES_TIMES_LIST = [
     "tbl_true_dates_times_pd",
     "tbl_true_dates_times_pl",
+    "tbl_true_dates_times_duckdb",
 ]
 
 
@@ -199,6 +200,12 @@ def tbl_dates_times_text_duckdb():
     return ibis.connect(f"duckdb://{file_path}").table("tbl_dates_times_text")
 
 
+@pytest.fixture
+def tbl_true_dates_times_duckdb():
+    file_path = pathlib.Path.cwd() / "tests" / "tbl_files" / "tbl_true_dates_times.ddb"
+    return ibis.connect(f"duckdb://{file_path}").table("tbl_true_dates_times")
+
+
 @pytest.fixture
 def tbl_sqlite():
     file_path = pathlib.Path.cwd() / "tests" / "tbl_files" / "tbl_xyz.sqlite"
@@ -4105,13 +4112,6 @@ def test_col_schema_match_columns_only():
 
 @pytest.mark.parametrize("tbl_fixture", TBL_TRUE_DATES_TIMES_LIST)
 def test_date_validation_across_cols(request, tbl_fixture):
-    # {
-    #     "date_1": pd.to_datetime(["2021-01-01", "2021-02-01"]),
-    #     "date_2": pd.to_datetime(["2021-02-01", "2021-03-01"]),
-    #     "dttm_1": pd.to_datetime(["2021-01-01 02:30:00", "2021-02-01 02:30:00"]),
-    #     "dttm_2": pd.to_datetime(["2021-02-01 03:30:00", "2021-03-01 03:30:00"]),
-    # }
-
     tbl = request.getfixturevalue(tbl_fixture)
 
     assert (
@@ -4183,13 +4183,6 @@ def test_date_validation_across_cols(request, tbl_fixture):
 
 @pytest.mark.parametrize("tbl_fixture", TBL_TRUE_DATES_TIMES_LIST)
 def test_datetime_validation_across_cols(request, tbl_fixture):
-    # {
-    #     "date_1": pd.to_datetime(["2021-01-01", "2021-02-01"]),
-    #     "date_2": pd.to_datetime(["2021-02-01", "2021-03-01"]),
-    #     "dttm_1": pd.to_datetime(["2021-01-01 02:30:00", "2021-02-01 02:30:00"]),
-    #     "dttm_2": pd.to_datetime(["2021-02-01 03:30:00", "2021-03-01 03:30:00"]),
-    # }
-
     tbl = request.getfixturevalue(tbl_fixture)
 
     assert (