✅ Remove xfail for replace config tests (#44)

pydantic · Jun 30, 2023 · 0cd8d34 · 0cd8d34
1 parent 3508bfb
commit 0cd8d34
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/unit/test_replace_config.py b/tests/unit/test_replace_config.py
@@ -130,7 +130,7 @@ class Config(SuperConfig):
         """
         self.assertCodemod(before, after)
 
-    @pytest.mark.xfail(reason="Not implemented yet")
+    @pytest.mark.xfail(reason="Comments inside Config are swallowed.")
     def test_inner_comments(self) -> None:
         before = """
         from pydantic import BaseModel
@@ -176,7 +176,6 @@ class Config(SuperConfig):
         """
         self.assertCodemod(before, after)
 
-    @pytest.mark.xfail(reason="Not implemented yet")
     def test_extra_enum(self) -> None:
         before = """
         from pydantic import BaseModel, Extra
@@ -186,7 +185,7 @@ class Config:
                 extra = Extra.allow
         """
         after = """
-        from pydantic import BaseModel
+        from pydantic import ConfigDict, BaseModel
 
         class Potato(BaseModel):
             model_config = ConfigDict(extra="allow")