Remove deprecation warning from test_help

pydantic · Dec 4, 2023 · f5fc834 · f5fc834
1 parent 0ed195a
commit f5fc834
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 18 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -2,7 +2,6 @@
 import platform
 import re
 import sys
-import warnings
 from collections import defaultdict
 from copy import deepcopy
 from dataclasses import dataclass
@@ -2978,12 +2977,9 @@ class Bar(BaseModel):
 
 
 def test_help(create_module):
-    # since pydoc/help access all attributes to generate their documentation,
-    # this triggers the deprecation warnings.
-    with warnings.catch_warnings():
-        module = create_module(
-            # language=Python
-            """
+    module = create_module(
+        # language=Python
+        """
 import pydoc
 
 from pydantic import BaseModel
@@ -2994,8 +2990,7 @@ class Model(BaseModel):
 
 help_result_string = pydoc.render_doc(Model)
 """
-        )
-    warnings.simplefilter('error')
+    )
     assert 'class Model' in module.help_result_string
 
 

diff --git a/tests/test_root_model.py b/tests/test_root_model.py
@@ -1,5 +1,4 @@
 import pickle
-import warnings
 from datetime import date, datetime
 from typing import Any, Dict, List, Optional, Union
 
@@ -592,21 +591,17 @@ class Model(RootModel):
 
 
 def test_help(create_module):
-    # since pydoc/help access all attributes to generate their documentation,
-    # this triggers the deprecation warnings.
-    with warnings.catch_warnings():
-        module = create_module(
-            # language=Python
-            """
+    module = create_module(
+        # language=Python
+        """
 import pydoc
 
 from pydantic import RootModel
 
 
 help_result_string = pydoc.render_doc(RootModel)
 """
-        )
-    warnings.simplefilter('error')
+    )
     assert 'class RootModel' in module.help_result_string