From d1631ecd8ec2947c4bd3aed05f5d23a7afa23e65 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebasti=C3=A1n=20Ram=C3=ADrez?= Date: Thu, 18 Apr 2024 01:22:26 -0500 Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20Fix=20tests=20for=20CLI,=20const=5F?= =?UTF-8?q?to=5Fliteral=20with=20Pydantic=20model=20and=20Enum=20with=20th?= =?UTF-8?q?e=20same=20class=20name?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/integration/cases/replace_validator.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/integration/cases/replace_validator.py b/tests/integration/cases/replace_validator.py index 82009b5..97914be 100644 --- 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", ], ), ),