diff --git a/pyproject.toml b/pyproject.toml index e85cce3..7d3a9b1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -24,7 +24,7 @@ dependencies = [ ] [project.optional-dependencies] airflow = ["apache-airflow"] -test = ["testcontainers>=3.7,<4.9", "apache-airflow", "pylint"] +test = ["testcontainers>=3.7,<4.9", "apache-airflow", "pylint", "dlt[duckdb]"] [tool.setuptools] include-package-data = true diff --git a/tests/resources/dbttest/models/my_dlt_import_model.py b/tests/resources/dbttest/models/my_dlt_import_model.py new file mode 100644 index 0000000..e70e106 --- /dev/null +++ b/tests/resources/dbttest/models/my_dlt_import_model.py @@ -0,0 +1,11 @@ +@dlt.resource( + columns={"event_tstamp": {"data_type": "timestamp", "timezone": False}}, + primary_key="event_id", +) +def events(): + yield [{"event_id": 1, "event_tstamp": "2024-07-30T10:00:00.123+00:00"}] + + +def model(dbt, session): + print(session) + print(type(session))