From 74ffbcb85c23895af1cdc4b40f804a17886e2dbf Mon Sep 17 00:00:00 2001 From: Guilherme Martins Crocetti Date: Mon, 10 Aug 2020 23:47:53 -0300 Subject: [PATCH] converters(test): add import logic --- tests/unit/test_converters.py | 30 +++++++++++++++++++----------- 1 file changed, 19 insertions(+), 11 deletions(-) diff --git a/tests/unit/test_converters.py b/tests/unit/test_converters.py index 5af8939b..29836319 100644 --- a/tests/unit/test_converters.py +++ b/tests/unit/test_converters.py @@ -1,9 +1,16 @@ # Standard library imports import typing +import sys # Third-party imports import marshmallow -import pydantic + +try: + import pydantic +except ImportError: + if sys.version_info >= (3, 6): + raise + import pytest # Local imports @@ -451,18 +458,19 @@ def test_dict_converter(self): assert output == "1" -@pytest.fixture -def pydantic_model_mock(mocker): - class Model(pydantic.BaseModel): - def __new__(cls, *args, **kwargs): - return model - - model = mocker.Mock(spec=Model) - - return model, Model +@pytest.mark.skipif( + sys.version_info < (3, 6), reason="requires python3.6 or higher" +) +class TestPydanticConverter(object): + @pytest.fixture + def pydantic_model_mock(self, mocker): + class Model(pydantic.BaseModel): + def __new__(cls, *args, **kwargs): + return model + model = mocker.Mock(spec=Model) + return model, Model -class TestPydanticConverter(object): def test_init_without_pydantic(self, mocker): mocker.patch.object( converters.PydanticConverter,