!squash

Author: tony

src/unihan_etl/validator.py

@@ -1,5 +1,4 @@
-import typing as t
-
+"""Experimental pydantic models for unihan data."""
 import pydantic
 
 from unihan_etl.expansion import expand_kTGHZ2013
@@ -19,7 +18,10 @@ class kTGHZ2013Location(pydantic.BaseModel):
     # 0 for a main entry and greater than 0 for a parenthesized or bracketed variant #
     # of the main entry
     entry_type: int = pydantic.Field(
-        description="0 for a main entry and greater than 0 for a parenthesized or bracketed variant of the main entry"
+        description=(
+            "0 for a main entry and greater than 0 for a parenthesized or bracketed "
+            + "variant of the main entry"
+        )
     )
 
 
@@ -64,16 +66,6 @@ def from_string(cls, value: str) -> "kTGHZ2013":
                     for out in outs
                 ],
             )
-            #     reading=out["reading"],
-            #     locations=[
-            #         kTGHZ2013Location(
-            #             page=location["page"],
-            #             position=location["position"],
-            #             entry_type=location["entry_type"],
-            #         )
-            #         for location in out["locations"]
-            #     ],
-            # )
         elif isinstance(value, dict):
             return pydantic.parse_obj_as(cls, value)
         raise pydantic.ValidationError("Invalid input for kTGHZ2013 model.")  # noqa: TRY003

tests/test_validator.py

@@ -1,15 +1,14 @@
 """Test expansion of multi-value fields in UNIHAN."""
 import typing as t
 
-import pytest
-
-from unihan_etl import constants, validator
+from unihan_etl import validator
 
 if t.TYPE_CHECKING:
-    from typing_extensions import TypeAlias
+    pass
 
 
 def test_kTGHZ2013() -> None:
+    """Example of kTGHZ2013 being parsed via pydantic."""
     model = validator.kTGHZ2013.from_string("U+3447	kTGHZ2013	482.140:zhòu")
     assert model.ucn == "U+3447"