mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-05-06 22:12:12 +08:00
9f0d592839
* remove python 3.9 support and code * black * more cli lint * more linting * more lint * fix for tests * docstring grammar police * add lock to to build function to avoid async import race conditions * grammar police * lots more linting * relock
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
"""Custom pytest configuration for the extensions."""
|
|
|
|
import pytest
|
|
from openbb_core.app.router import CommandMap
|
|
|
|
from .utils.helpers import list_openbb_extensions
|
|
|
|
cm = CommandMap()
|
|
commands = list(cm.map.keys())
|
|
|
|
# ruff: noqa: SIM114
|
|
|
|
|
|
def parametrize(argnames: str, argvalues: list, **kwargs):
|
|
"""Apply a custom parametrize decorator that filters test cases based on the environment."""
|
|
|
|
routers, providers, obbject_ext = list_openbb_extensions()
|
|
|
|
def decorator(function):
|
|
"""Patch the pytest.mark.parametrize decorator."""
|
|
filtered_argvalues: list[dict] = []
|
|
name = function.__name__.split("_")[1]
|
|
# This is a patch to handle the charting extension name
|
|
extension_name = "openbb_" + name if name == "charting" else name
|
|
function_name = "/" + "/".join(function.__name__.split("_")[1:])
|
|
# this handles edge cases where the function name has an underscore
|
|
function_name_v2 = (
|
|
function_name.rsplit("/", 1)[0] + "_" + function_name.rsplit("/", 1)[1]
|
|
)
|
|
function_name_v3 = (
|
|
function_name.rsplit("/", 2)[0]
|
|
+ "_"
|
|
+ function_name.rsplit("/", 2)[1]
|
|
+ "_"
|
|
+ function_name.rsplit("/", 1)[1].replace("/", "_")
|
|
)
|
|
if extension_name in routers | obbject_ext:
|
|
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)
|
|
elif "provider" not in args and function_name_v2 in commands:
|
|
# Handle edge case
|
|
filtered_argvalues.append(args)
|
|
elif "provider" not in args and function_name_v3 in commands:
|
|
# Handle edge case
|
|
filtered_argvalues.append(args)
|
|
elif extension_name in obbject_ext:
|
|
filtered_argvalues.append(args)
|
|
|
|
# If filtered_argvalues is empty, pytest will skip the test!
|
|
return pytest.mark.parametrize(argnames, filtered_argvalues, **kwargs)(
|
|
function
|
|
)
|
|
|
|
return pytest.mark.skip(function)
|
|
|
|
return decorator
|