mirror of
https://github.com/mihakralj/QuanTAlib.git
synced 2026-07-28 09:47:43 +00:00
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
|
"""test_category_labels.py — Verify consistent category labels across all modules."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import importlib
|
||
|
|
import inspect
|
||
|
|
import re
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
CATEGORY_MODULES = {
|
||
|
|
"quantalib.channels": "channels",
|
||
|
|
"quantalib.core": "core",
|
||
|
|
"quantalib.cycles": "cycles",
|
||
|
|
"quantalib.dynamics": "dynamics",
|
||
|
|
"quantalib.errors": "errors",
|
||
|
|
"quantalib.filters": "filters",
|
||
|
|
"quantalib.momentum": "momentum",
|
||
|
|
"quantalib.numerics": "numerics",
|
||
|
|
"quantalib.oscillators": "oscillators",
|
||
|
|
"quantalib.reversals": "reversals",
|
||
|
|
"quantalib.statistics": "statistics",
|
||
|
|
"quantalib.trends_fir": "trends_fir",
|
||
|
|
"quantalib.trends_iir": "trends_iir",
|
||
|
|
"quantalib.volatility": "volatility",
|
||
|
|
"quantalib.volume": "volume",
|
||
|
|
}
|
||
|
|
|
||
|
|
# Regex to match _wrap(..., "CATEGORY", ...) or _wrap_multi(..., "CATEGORY", ...)
|
||
|
|
LABEL_PATTERN = re.compile(r'_wrap(?:_multi)?\(.*?,\s*"([^"]+)",\s*(?:offset|[\w]+)\)')
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("modname,expected", CATEGORY_MODULES.items(), ids=lambda x: x.split(".")[-1] if "." in x else x)
|
||
|
|
def test_category_labels_lowercase(modname: str, expected: str) -> None:
|
||
|
|
"""Category labels passed to _wrap/_wrap_multi must be lowercase."""
|
||
|
|
mod = importlib.import_module(modname)
|
||
|
|
source_file = inspect.getfile(mod)
|
||
|
|
with open(source_file, "r", encoding="utf-8") as f:
|
||
|
|
source = f.read()
|
||
|
|
|
||
|
|
# Find all category labels in source
|
||
|
|
labels = LABEL_PATTERN.findall(source)
|
||
|
|
bad = [lbl for lbl in labels if lbl != expected]
|
||
|
|
assert bad == [], (
|
||
|
|
f"{modname}: expected category label '{expected}', found non-matching: {set(bad)}"
|
||
|
|
)
|