diff options
author | Henrique Joaquim <h.joaquim@campus.fct.unl.pt> | 2024-02-26 10:30:34 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-02-26 10:30:34 +0000 |
commit | 1d29ddcb861c19f8b06cef5e9812e5703c689476 (patch) | |
tree | 4175334dad33a3c3f16010e5ae3c978e1e0686f5 | |
parent | 1cbe34529eb6aeb0fcf0e13d145f896e8c38de16 (diff) |
remove unnecessary function (#6125)
-rw-r--r-- | openbb_platform/providers/tests/test_provider_fetcher.py | 17 |
1 files changed, 4 insertions, 13 deletions
diff --git a/openbb_platform/providers/tests/test_provider_fetcher.py b/openbb_platform/providers/tests/test_provider_fetcher.py index 6aaf4635315..bfe209d0ec9 100644 --- a/openbb_platform/providers/tests/test_provider_fetcher.py +++ b/openbb_platform/providers/tests/test_provider_fetcher.py @@ -14,15 +14,6 @@ from providers.tests.utils.unit_tests_generator import ( ) -def get_providers() -> Dict[str, Provider]: - """Get the providers from the provider registry.""" - providers: Dict[str, Provider] = {} - registry = RegistryLoader.from_extensions() - for provider_name, provider_cls in registry.providers.items(): - providers[provider_name] = provider_cls - return providers - - def get_provider_test_files(provider: Provider): """Given a provider, return the path to the test file.""" fetchers_dict = provider.fetcher_dict @@ -40,11 +31,12 @@ def get_provider_test_files(provider: Provider): class ProviderFetcherTest(unittest.TestCase): """Tests for providers and fetchers.""" + providers: Dict[str, Provider] = RegistryLoader.from_extensions().providers + def test_provider_w_tests(self): """Test the provider fetchers and ensure all providers have tests.""" - providers = get_providers() - for provider_name, provider_cls in providers.items(): + for provider_name, provider_cls in self.providers.items(): with self.subTest(i=provider_name): path = get_provider_test_files(provider_cls) @@ -52,13 +44,12 @@ class ProviderFetcherTest(unittest.TestCase): def test_provider_fetchers_w_tests(self): """Ensure all the fetchers in each provider have tests.""" - providers = get_providers() provider_fetchers = get_provider_fetchers() for provider_name, fetcher_dict in provider_fetchers.items(): for _, fetcher_cls in fetcher_dict.items(): - path = get_provider_test_files(providers[provider_name]) + path = get_provider_test_files(self.providers[provider_name]) # check that fetcher_cls is being instantiated in path with self.subTest(i=fetcher_cls): |