✅ Tweak test for coverage

7 months ago · e0d987fee6
1 changed files with 2 additions and 3 deletions
--- a/tests/test_union_body_discriminator_annotated.py
+++ b/tests/test_union_body_discriminator_annotated.py
@ -26,9 +26,8 @@ def client_fixture() -> TestClient:
        barks: float
    def get_pet_type(v):
-        if isinstance(v, dict):
+        assert isinstance(v, dict)
-            return v.get("pet_type", "")
+        return v.get("pet_type", "")
        return getattr(v, "pet_type", "")
    Pet = Annotated[
        Union[Annotated[Cat, Tag("cat")], Annotated[Dog, Tag("dog")]],