diff --git a/tests/functional/shell/test_script_subprocess.py b/tests/functional/shell/test_script_subprocess.py index 3bbc6d5..3255056 100644 --- a/tests/functional/shell/test_script_subprocess.py +++ b/tests/functional/shell/test_script_subprocess.py @@ -8,7 +8,10 @@ from typing import cast import pytest -from pytest_subtests import SubTests +try: + from pytest import Subtests as SubTests +except ImportError: + from pytest_subtests import SubTests from pytestshellutils.customtypes import EnvironDict from pytestshellutils.exceptions import FactoryTimeout diff --git a/tests/unit/utils/processes/test_processresult.py b/tests/unit/utils/processes/test_processresult.py index 3ab90fd..5c88a52 100644 --- a/tests/unit/utils/processes/test_processresult.py +++ b/tests/unit/utils/processes/test_processresult.py @@ -8,7 +8,10 @@ import textwrap import pytest -from pytest_subtests import SubTests +try: + from pytest import Subtests as SubTests +except ImportError: + from pytest_subtests import SubTests from pytestshellutils.utils.processes import ProcessResult