✅ Fix tests for CLI, const_to_literal with Pydantic model and Enum wi…

…th the same class name
pydantic · Apr 18, 2024 · d1631ec · d1631ec
1 parent 969f63f
commit d1631ec
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/integration/cases/replace_validator.py b/tests/integration/cases/replace_validator.py
@@ -54,12 +54,12 @@
                 "from pydantic import BaseModel, Field",
                 "",
                 "",
-                "class A(str, Enum):",
+                "class E(str, Enum):",
                 "    a = 'a'",
                 "    b = 'b'",
                 "",
                 "class A(BaseModel):",
-                "    a: A = Field(A.a, const=True)",
+                "    a: E = Field(E.a, const=True)",
             ],
         ),
         expected=File(
@@ -70,12 +70,12 @@
                 "from typing import Literal",
                 "",
                 "",
-                "class A(str, Enum):",
+                "class E(str, Enum):",
                 "    a = 'a'",
                 "    b = 'b'",
                 "",
                 "class A(BaseModel):",
-                "    a: Literal[A.a] = A.a",
+                "    a: Literal[E.a] = E.a",
             ],
         ),
     ),