mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 21:12:16 +00:00
13283fbfce
* BIGGGG LINTING * fixing lints * fixing lints * black * very ruff * no export * fix hedge_view again * lints * platform lints * lints * black * black it @hjoaquim * fix some more linting --------- Co-authored-by: hjoaquim <h.joaquim@campus.fct.unl.pt>
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
"""Custom pytest configuration for the extensions."""
|
|
|
|
from typing import Dict, List
|
|
|
|
import pytest
|
|
from openbb_core.app.router import CommandMap
|
|
|
|
from extensions.tests.utils.helpers import list_openbb_extensions
|
|
|
|
cm = CommandMap()
|
|
commands = list(cm.map.keys())
|
|
|
|
|
|
def parametrize(argnames: str, argvalues: List[Dict], **kwargs):
|
|
"""Custom parametrize decorator that filters test cases based on the environment."""
|
|
|
|
extensions, providers = list_openbb_extensions()
|
|
|
|
def decorator(function):
|
|
"""Patch the pytest.mark.parametrize decorator."""
|
|
filtered_argvalues: List[Dict] = []
|
|
extension_name = function.__name__.split("_")[1]
|
|
function_name = "/" + "/".join(function.__name__.split("_")[1:])
|
|
|
|
if extension_name in extensions:
|
|
for args in argvalues:
|
|
if "provider" in args and args["provider"] in providers:
|
|
filtered_argvalues.append(args)
|
|
elif "provider" not in args and function_name in commands:
|
|
# Run the standard test case
|
|
filtered_argvalues.append(args)
|
|
return pytest.mark.parametrize(argnames, filtered_argvalues, **kwargs)(
|
|
function
|
|
)
|
|
|
|
return pytest.mark.skip(function)
|
|
|
|
return decorator
|