diff --git a/tests/unit/wiring/package/__init__.py b/tests/unit/wiring/package/__init__.py index 83d62e58..bdc34ed2 100644 --- a/tests/unit/wiring/package/__init__.py +++ b/tests/unit/wiring/package/__init__.py @@ -1,8 +1,11 @@ -from dependency_injector.wiring import Provide +import sys -from ..container import Container -from ..service import Service +if sys.version_info >= (3, 6): + from dependency_injector.wiring import Provide + + from ..container import Container + from ..service import Service -def test_package_function(service: Service = Provide[Container.service]): - return service + def test_package_function(service: Service = Provide[Container.service]): + return service diff --git a/tests/unit/wiring/package/subpackage/__init__.py b/tests/unit/wiring/package/subpackage/__init__.py index e73dd8ce..b2d3278b 100644 --- a/tests/unit/wiring/package/subpackage/__init__.py +++ b/tests/unit/wiring/package/subpackage/__init__.py @@ -1,8 +1,11 @@ -from dependency_injector.wiring import Provide +import sys -from ...container import Container -from ...service import Service +if sys.version_info >= (3, 6): + from dependency_injector.wiring import Provide + + from ...container import Container + from ...service import Service -def test_package_function(service: Service = Provide[Container.service]): - return service + def test_package_function(service: Service = Provide[Container.service]): + return service