diff --git a/tests/test_color_compat.py b/tests/test_color_compat.py index 2fcdd298c..56dfe135f 100644 --- a/tests/test_color_compat.py +++ b/tests/test_color_compat.py @@ -1,4 +1,5 @@ import warnings + import pydantic try: @@ -6,12 +7,18 @@ try: except ImportError: from pydantic.color import Color # triggers DeprecationWarning on v1 + class Model(pydantic.BaseModel): c: Color + def test_color_deprecation_warning(): with warnings.catch_warnings(record=True) as w: warnings.simplefilter("always") - Model(c="#FF0000") - dep_warnings = [warn for warn in w if issubclass(warn.category, DeprecationWarning)] + # instantiate to trigger warning + Model(c=Color("#FF0000")) + + dep_warnings = [ + warn for warn in w if issubclass(warn.category, DeprecationWarning) + ] assert len(dep_warnings) == 0, "DeprecationWarning raised! Fixed in this PR"