Function test_datetimes

dask/dataframe/io/tests/test_sql.py:348–367

()

def test_datetimes():
    import datetime

    now = datetime.datetime.now()
    d = datetime.timedelta(seconds=1)
    df = pd.DataFrame(
        {"a": list("ghjkl"), "b": [now + i * d for i in range(2, -3, -1)]}
    )
    with tmpfile() as f:
        uri = f"sqlite:///{f}"
        df.to_sql("test", uri, index=False, if_exists="replace")
        data = read_sql_table("test", uri, npartitions=2, index_col="b")
        assert data.index.dtype.kind == "M"
        assert data.divisions[0] == df.b.min()
        df2 = df.set_index("b")
        assert_eq(
            data.map_partitions(lambda x: x.sort_index()),
            df2.sort_index(),
            check_dtype=False,
        )


def test_extra_connection_engine_keywords(caplog, db):