From 43c31c5dc2bf596176762e4e7c01df35d80459fe Mon Sep 17 00:00:00 2001 From: Matt Clay Date: Fri, 26 May 2023 19:25:26 -0700 Subject: [PATCH] Use isinstance in unit tests for type checks (#80906) --- test/units/mock/loader.py | 2 +- test/units/module_utils/basic/test_safe_eval.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test/units/mock/loader.py b/test/units/mock/loader.py index 4bc7eab5a3f..82e6dbac8e2 100644 --- a/test/units/mock/loader.py +++ b/test/units/mock/loader.py @@ -30,7 +30,7 @@ class DictDataLoader(DataLoader): def __init__(self, file_mapping=None): file_mapping = {} if file_mapping is None else file_mapping - assert type(file_mapping) == dict + assert isinstance(file_mapping, dict) super(DictDataLoader, self).__init__() diff --git a/test/units/module_utils/basic/test_safe_eval.py b/test/units/module_utils/basic/test_safe_eval.py index e8538ca911d..fdaab18abfc 100644 --- a/test/units/module_utils/basic/test_safe_eval.py +++ b/test/units/module_utils/basic/test_safe_eval.py @@ -67,4 +67,4 @@ def test_invalid_strings_with_exceptions(am, code, expected, exception): if exception is None: assert res[1] == exception else: - assert type(res[1]) == exception + assert isinstance(res[1], exception)