diff options
author | Theodore Aptekarev <aptekarev@gmail.com> | 2023-09-20 10:46:23 +0200 |
---|---|---|
committer | Theodore Aptekarev <aptekarev@gmail.com> | 2023-09-20 10:46:23 +0200 |
commit | ac5bcc5e9a7fd8fd87369828d10fb610cb2ec302 (patch) | |
tree | eb2682727d0276e88c4dcc6ca1a6b64b5582656f | |
parent | 6ed837d24c117b09b68f6fd84522886fa8c62bea (diff) |
Fix linting errors
-rw-r--r-- | openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py | 30 |
1 files changed, 13 insertions, 17 deletions
diff --git a/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py b/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py index e71e4afb35c..1abb5f23853 100644 --- a/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py +++ b/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py @@ -17,9 +17,11 @@ class classproperty: """Class property decorator.""" def __init__(self, f): + """Initialize decorator.""" self.f = f def __get__(self, obj, owner): + """Get the property.""" return self.f(owner) @@ -105,18 +107,16 @@ class Fetcher(Generic[Q, R]): # Query Assertions assert query assert issubclass(type(query), cls.query_params_type) - assert all([getattr(query, key) == value for key, value in test_params.items()]) + assert all(getattr(query, key) == value for key, value in test_params.items()) # Data Assertions assert data - is_list = type(data) == list + is_list = isinstance(data, list) if is_list: assert all( - [ - field in data[0] - for field in cls.data_type.__fields__ - if field in data[0] - ] + field in data[0] + for field in cls.data_type.__fields__ + if field in data[0] ) # This makes sure that the data is not transformed yet so that the # pipeline is implemented correctly. We can remove this assertion if we @@ -124,7 +124,7 @@ class Fetcher(Generic[Q, R]): assert issubclass(type(data[0]), cls.data_type) is False else: assert all( - [field in data for field in cls.data_type.__fields__ if field in data] + field in data for field in cls.data_type.__fields__ if field in data ) assert issubclass(type(data), cls.data_type) is False @@ -133,14 +133,12 @@ class Fetcher(Generic[Q, R]): # Transformed Data Assertions assert transformed_data - is_list = type(transformed_data) == list + is_list = isinstance(transformed_data, list) if is_list: assert len(transformed_data) > 0 # type: ignore assert all( - [ - field in transformed_data[0].__dict__ # type: ignore - for field in cls.return_type.__args__[0].__fields__ - ] + field in transformed_data[0].__dict__ # type: ignore + for field in cls.return_type.__args__[0].__fields__ ) assert issubclass(type(transformed_data[0]), cls.data_type) # type: ignore assert issubclass( @@ -149,10 +147,8 @@ class Fetcher(Generic[Q, R]): ) else: assert all( - [ - field in transformed_data.__dict__ - for field in cls.return_type.__fields__ - ] + field in transformed_data.__dict__ + for field in cls.return_type.__fields__ ) assert issubclass(type(transformed_data), cls.data_type) assert issubclass(type(transformed_data), cls.return_type) |