From 078c8dc8f1fd3a8361114d45b7f222fc1f0c4b87 Mon Sep 17 00:00:00 2001 From: Arthur Meyre Date: Wed, 4 Aug 2021 17:23:26 +0200 Subject: [PATCH] fix: conform to more reasonable style to check type of object (#78) --- hdk/common/data_types/dtypes_helpers.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/hdk/common/data_types/dtypes_helpers.py b/hdk/common/data_types/dtypes_helpers.py index 85f6ad2f7..a7036b950 100644 --- a/hdk/common/data_types/dtypes_helpers.py +++ b/hdk/common/data_types/dtypes_helpers.py @@ -6,7 +6,7 @@ from .base import BaseDataType from .integers import Integer from .values import BaseValue, ClearValue, EncryptedValue -INTEGER_TYPES = set([Integer]) +INTEGER_TYPES = (Integer,) def value_is_encrypted_integer(value_to_check: BaseValue) -> bool: @@ -18,9 +18,8 @@ def value_is_encrypted_integer(value_to_check: BaseValue) -> bool: Returns: bool: True if the passed value_to_check is an encrypted value of type Integer """ - return ( - isinstance(value_to_check, EncryptedValue) - and type(value_to_check.data_type) in INTEGER_TYPES + return isinstance(value_to_check, EncryptedValue) and isinstance( + value_to_check.data_type, INTEGER_TYPES )