test: add test for enum reflection

Mause · Nov 16, 2022 · e478a46 · e478a46
1 parent 5e4549e
commit e478a46
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/duckdb_engine/tests/test_basic.py b/duckdb_engine/tests/test_basic.py
@@ -234,6 +234,22 @@ def test_table_reflect(session: Session, engine: Engine) -> None:
     insp.reflecttable(user_table, None)
 
 
+def test_enum_reflection(session: Session, engine: Engine) -> None:
+    importorskip("duckdb", "0.6.1-dev0")
+
+    sql = [
+        "CREATE TYPE enum_t AS ENUM('a', 'b');",
+        "CREATE TABLE tmp (enum_col enum_t);",
+        "INSERT INTO tmp VALUES ('b');",
+        "INSERT INTO tmp VALUES ('a');",
+    ]
+    for s in sql:
+        session.execute(text(s))
+
+    tmp = Table("tmp", MetaData())
+    assert inspect(engine).reflecttable(tmp, include_columns=True)
+
+
 def test_fetch_df_chunks() -> None:
     import duckdb