diff --git a/validations_libs/tests/test_ansible.py b/validations_libs/tests/test_ansible.py index 89225f4..5925892 100644 --- a/validations_libs/tests/test_ansible.py +++ b/validations_libs/tests/test_ansible.py @@ -13,7 +13,10 @@ # under the License. # -from unittest import mock +try: + from unittest import mock +except ImportError: + import mock from unittest import TestCase from ansible_runner import Runner diff --git a/validations_libs/tests/test_utils.py b/validations_libs/tests/test_utils.py index fb74853..b1a03be 100644 --- a/validations_libs/tests/test_utils.py +++ b/validations_libs/tests/test_utils.py @@ -13,7 +13,10 @@ # under the License. # -from unittest import mock +try: + from unittest import mock +except ImportError: + import mock from unittest import TestCase from validations_libs import utils diff --git a/validations_libs/tests/test_validations_list.py b/validations_libs/tests/test_validations_list.py index 3e30cf0..a5cf5a5 100644 --- a/validations_libs/tests/test_validations_list.py +++ b/validations_libs/tests/test_validations_list.py @@ -13,7 +13,10 @@ # under the License. # -from unittest import mock +try: + from unittest import mock +except ImportError: + import mock from unittest import TestCase from validations_libs.tests import fakes diff --git a/validations_libs/tests/test_validations_run.py b/validations_libs/tests/test_validations_run.py index 2749c0d..416a141 100644 --- a/validations_libs/tests/test_validations_run.py +++ b/validations_libs/tests/test_validations_run.py @@ -13,7 +13,10 @@ # under the License. # -from unittest import mock +try: + from unittest import mock +except ImportError: + import mock from unittest import TestCase from validations_libs.tests import fakes diff --git a/validations_libs/tests/test_validations_show.py b/validations_libs/tests/test_validations_show.py index 516a20b..9f4a44d 100644 --- a/validations_libs/tests/test_validations_show.py +++ b/validations_libs/tests/test_validations_show.py @@ -13,7 +13,10 @@ # under the License. # -from unittest import mock +try: + from unittest import mock +except ImportError: + import mock from unittest import TestCase from validations_libs.tests import fakes