ci: Add tests for serialization and printing of dy.Enum columns (#147)

Co-authored-by: Andreas Albert <[email protected]>

Author: MoritzPotthoffQC

tests/schema/test_repr.py

@@ -52,3 +52,14 @@ def test_repr_with_rules() -> None:
             - "my_rule": [(col("a")) < (dyn int: 100)]
             - "my_group_rule": [(col("a").sum()) > (dyn int: 50)] grouped by ['a']
         """)
+
+
+def test_repr_enum() -> None:
+    class SchemaNoRules(dy.Schema):
+        a = dy.Enum(["a"])
+
+    assert repr(SchemaNoRules) == textwrap.dedent("""\
+        [Schema "SchemaNoRules"]
+          Columns:
+            - "a": Enum(categories=['a'], nullable=True)
+        """)

tests/schema/test_serialization.py

@@ -40,6 +40,7 @@ def test_simple_serialization() -> None:
         create_schema("test", {"a": dy.Int64(check=[lambda expr: expr > 5])}),
         create_schema("test", {"a": dy.Int64(check={"x": lambda expr: expr > 5})}),
         create_schema("test", {"a": dy.Int64(alias="foo")}),
+        create_schema("test", {"a": dy.Enum(["a"])}),
         create_schema("test", {"a": dy.Decimal(scale=2, min=Decimal("1.5"))}),
         create_schema("test", {"a": dy.Date(min=dt.date(2020, 1, 1))}),
         create_schema("test", {"a": dy.Datetime(min=dt.datetime(2020, 1, 1))}),