mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-31 14:31:20 +00:00
8ea35b0bd0
* refactor imf extension for V5 * codespell * fix some python 3.10 items * return self on copy of metadata object
115 lines
3.5 KiB
Python
115 lines
3.5 KiB
Python
"""Shared pytest fixtures for the IMF provider."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _reset_imf_metadata_singleton() -> Any:
|
|
"""Drop the ``ImfMetadata`` singleton before and after each test."""
|
|
from openbb_imf.utils.metadata import ImfMetadata
|
|
|
|
ImfMetadata._reset()
|
|
yield
|
|
ImfMetadata._reset()
|
|
|
|
|
|
def _reset_alru(fn: Any) -> None:
|
|
"""Clear an ``alru_cache`` wrapper and reset its event-loop binding."""
|
|
if hasattr(fn, "cache_clear"):
|
|
fn.cache_clear()
|
|
for attr in ("_LRUCacheWrapper__first_loop", "_LRUCacheWrapper__warned_loop_reset"):
|
|
if hasattr(fn, attr):
|
|
setattr(fn, attr, None if "first_loop" in attr else False)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _clear_alru_caches() -> Any:
|
|
"""Clear every ``alru_cache``-wrapped helper between tests."""
|
|
from openbb_imf.utils import port_watch_helpers as pwh
|
|
|
|
targets = [
|
|
pwh.get_daily_chokepoint_data,
|
|
pwh.get_all_daily_chokepoint_activity_data,
|
|
pwh.get_all_daily_port_activity_data,
|
|
pwh.get_daily_port_activity_data,
|
|
pwh.list_ports,
|
|
pwh.get_country_daily_activity,
|
|
pwh.get_monthly_trade,
|
|
pwh.get_container_metrics,
|
|
pwh.get_disruption_events,
|
|
pwh.get_disruption_sankey_edges,
|
|
pwh.get_container_port_name_map,
|
|
]
|
|
for fn in targets:
|
|
_reset_alru(fn)
|
|
yield
|
|
for fn in targets:
|
|
_reset_alru(fn)
|
|
|
|
|
|
@pytest.fixture
|
|
def empty_meta(monkeypatch: pytest.MonkeyPatch) -> Any:
|
|
"""Fresh ``ImfMetadata`` singleton with cache loading disabled."""
|
|
from openbb_imf.utils.metadata import ImfMetadata
|
|
|
|
ImfMetadata._reset()
|
|
monkeypatch.setattr(ImfMetadata, "_load_from_cache", lambda self: True)
|
|
instance = ImfMetadata()
|
|
yield instance
|
|
ImfMetadata._reset()
|
|
|
|
|
|
@pytest.fixture
|
|
def seeded_meta(monkeypatch: pytest.MonkeyPatch) -> Any:
|
|
"""Fresh ``ImfMetadata`` with a minimal in-memory fixture catalog."""
|
|
from openbb_imf.utils.metadata import ImfMetadata
|
|
|
|
ImfMetadata._reset()
|
|
monkeypatch.setattr(ImfMetadata, "_load_from_cache", lambda self: True)
|
|
instance = ImfMetadata()
|
|
|
|
instance.dataflows = {
|
|
"TEST_DF": {
|
|
"id": "TEST_DF",
|
|
"agencyID": "IMF.STA",
|
|
"name": "Test Dataflow",
|
|
"description": "",
|
|
"structureRef": {"id": "DSD_TEST"},
|
|
}
|
|
}
|
|
instance.datastructures = {
|
|
"DSD_TEST": {
|
|
"id": "DSD_TEST",
|
|
"agencyID": "IMF.STA",
|
|
"dimensions": [
|
|
{"id": "COUNTRY", "position": "0", "conceptRef": {"id": "COUNTRY"}},
|
|
{"id": "FREQUENCY", "position": "1", "conceptRef": {"id": "FREQ"}},
|
|
{"id": "INDICATOR", "position": "2", "conceptRef": {"id": "INDICATOR"}},
|
|
],
|
|
}
|
|
}
|
|
instance._codelist_cache = {
|
|
"CL_COUNTRY": {
|
|
"USA": "United States",
|
|
"GBR": "United Kingdom",
|
|
"DEU": "Germany",
|
|
},
|
|
"CL_FREQ": {"A": "Annual", "Q": "Quarterly", "M": "Monthly"},
|
|
"CL_INDICATOR": {
|
|
"GDP": "Gross Domestic Product",
|
|
"CPI": "Consumer Price Index",
|
|
"POP": "Population",
|
|
},
|
|
}
|
|
instance._codelist_descriptions = {
|
|
"CL_COUNTRY": {"USA": "USA", "GBR": "UK", "DEU": "DE"},
|
|
"CL_FREQ": {"A": "Annual", "Q": "Quarterly", "M": "Monthly"},
|
|
"CL_INDICATOR": {"GDP": "GDP", "CPI": "CPI", "POP": "Population"},
|
|
}
|
|
yield instance
|
|
ImfMetadata._reset()
|