mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-31 22:41:19 +00:00
4400cd9a8f
* small changes - moving provider inside the core * uptade readme * lint * remove old provider readme * leftovers * refactor website examples * ruff * fix nox * relock * fix fails_to_deliver headers * fix for tests that have renamed endpoints * fix discovery * removed test for renamed endpoint * fix broken syntax - test_search_financial_attributes * fix cash/income/balance * fix derivatives tests * fix etf tests * fix news tests * fix regulators * black * wrong test param * sort equity gainers/losers by percent * platform markdown generation import * Hotfix/fix int tests core merge (#5779) * removing tests on old endpoint * removing visualization on old endpoint * fixing equity tests * fixing examples * fix news tests * fix index test * example * add comments * example --------- Co-authored-by: Diogo Sousa <montezdesousa@gmail.com> --------- Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com> Co-authored-by: Diogo Sousa <montezdesousa@gmail.com> Co-authored-by: Danglewood <85772166+deeleeramone@users.noreply.github.com> Co-authored-by: Pratyush Shukla <ps4534@nyu.edu>
67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
"""Test if providers and fetchers are covered by tests."""
|
|
import os
|
|
import unittest
|
|
from importlib import import_module
|
|
from typing import Dict
|
|
|
|
from openbb_core.provider.abstract.provider import Provider
|
|
from openbb_core.provider.registry import RegistryLoader
|
|
|
|
from providers.tests.utils.unit_tests_generator import (
|
|
check_pattern_in_file,
|
|
get_provider_fetchers,
|
|
)
|
|
|
|
|
|
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
|
|
fetcher_module_name = fetchers_dict[list(fetchers_dict.keys())[0]].__module__
|
|
parent_module = import_module(fetcher_module_name.split(".")[0])
|
|
parent_module_path = os.path.dirname(parent_module.__file__)
|
|
root_provider_path = os.path.dirname(parent_module_path)
|
|
provider_name = provider.name.lower()
|
|
|
|
return os.path.join(
|
|
root_provider_path, "tests", f"test_{provider_name}_fetchers.py"
|
|
)
|
|
|
|
|
|
class ProviderFetcherTest(unittest.TestCase):
|
|
"""Tests for providers and fetchers."""
|
|
|
|
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():
|
|
with self.subTest(i=provider_name):
|
|
path = get_provider_test_files(provider_cls)
|
|
|
|
self.assertTrue(os.path.exists(path))
|
|
|
|
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])
|
|
|
|
# check that fetcher_cls is being instantiated in path
|
|
with self.subTest(i=fetcher_cls):
|
|
self.assertTrue(
|
|
check_pattern_in_file(path, f"{fetcher_cls.__name__}()")
|
|
)
|