"""Test economy extension."""

import pytest
from openbb_core.app.model.obbject import OBBject


@pytest.fixture(scope="session")
def obb(pytestconfig):  # pylint: disable=inconsistent-return-statements
    """Fixture to setup obb."""

    if pytestconfig.getoption("markexpr") != "not integration":
        import openbb  # pylint: disable=import-outside-toplevel

        return openbb.obb


# pylint: disable=redefined-outer-name


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "nasdaq",
                "start_date": "2023-10-24",
                "end_date": "2023-11-03",
                "country": "united_states,japan",
            }
        ),
        (
            {
                "provider": "tradingeconomics",
                "start_date": "2023-01-01",
                "end_date": "2023-06-06",
                "country": "mexico,sweden",
                "importance": "low",
                "group": "gdp",
                "calendar_id": None,
            }
        ),
        (
            {
                "provider": "fmp",
                "start_date": "2023-10-24",
                "end_date": "2023-11-03",
            }
        ),
        (
            {
                "provider": "fred",
                "start_date": "2025-07-01",
                "end_date": "2025-07-02",
                "release_id": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_calendar(params, obb):
    """Test economy calendar."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.calendar(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        {
            "country": "spain",
            "transform": "yoy",
            "frequency": "annual",
            "harmonized": False,
            "start_date": "2020-01-01",
            "end_date": "2023-06-06",
            "provider": "fred",
        },
        {
            "country": "portugal,spain",
            "transform": "period",
            "frequency": "monthly",
            "harmonized": True,
            "start_date": "2023-01-01",
            "end_date": "2023-06-06",
            "provider": "fred",
        },
        {
            "country": "portugal,spain",
            "transform": "yoy",
            "frequency": "quarter",
            "harmonized": False,
            "start_date": "2020-01-01",
            "end_date": "2023-06-06",
            "provider": "oecd",
            "expenditure": "transport",
        },
        {
            "country": "portugal,spain",
            "transform": "yoy",
            "frequency": "quarter",
            "harmonized": False,
            "start_date": "2020-01-01",
            "end_date": "2023-06-06",
            "provider": "imf",
            "expenditure": "transport",
            "limit": None,
        },
    ],
)
@pytest.mark.integration
def test_economy_cpi(params, obb):
    """Test economy cpi."""
    result = obb.economy.cpi(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        ({"provider": "fmp"}),
    ],
)
@pytest.mark.integration
def test_economy_risk_premium(params, obb):
    """Test economy risk premium."""
    result = obb.economy.risk_premium(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states",
                "provider": "oecd",
                "frequency": "annual",
                "start_date": None,
                "end_date": None,
                "units": "volume",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_gdp_forecast(params, obb):
    """Test economy gdp forecast."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.gdp.forecast(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states",
                "start_date": "2023-01-01",
                "end_date": "2023-06-06",
                "provider": "econdb",
                "use_cache": False,
            }
        ),
        (
            {
                "country": "united_states",
                "provider": "oecd",
                "units": "level",
                "price_base": "volume",
                "frequency": "quarter",
                "start_date": "2023-01-01",
                "end_date": "2023-06-06",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_gdp_nominal(params, obb):
    """Test economy gdp nominal."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.gdp.nominal(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states",
                "frequency": "quarter",
                "start_date": "2023-01-01",
                "end_date": "2023-12-31",
                "provider": "oecd",
            }
        ),
        (
            {
                "country": "united_states",
                "provider": "econdb",
                "start_date": "2023-01-01",
                "end_date": "2023-12-31",
                "use_cache": False,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_gdp_real(params, obb):
    """Test economy gdp real."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.gdp.real(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "report_type": "summary",
                "frequency": "monthly",
                "country": None,
                "provider": "ecb",
            }
        ),
        (
            {
                "report_type": "direct_investment",
                "frequency": "monthly",
                "country": None,
                "provider": "ecb",
            }
        ),
        (
            {
                "report_type": "main",
                "frequency": "quarterly",
                "country": "united_states",
                "provider": "ecb",
            }
        ),
        (
            {
                "country": "united_states",
                "start_date": None,
                "end_date": None,
                "provider": "fred",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_balance_of_payments(params, obb):
    """Test economy balance of payments."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.balance_of_payments(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "query": "GDP*",
                "search_type": "series_id",
                "release_id": None,
                "offset": 0,
                "limit": 10,
                "order_by": "observation_end",
                "sort_order": "desc",
                "filter_variable": None,
                "filter_value": None,
                "tag_names": None,
                "exclude_tag_names": None,
                "series_id": None,
                "provider": "fred",
            }
        ),
        (
            {
                "query": None,
                "search_type": "release",
                "release_id": None,
                "offset": None,
                "limit": None,
                "order_by": "observation_end",
                "sort_order": "desc",
                "filter_variable": None,
                "filter_value": None,
                "tag_names": None,
                "exclude_tag_names": None,
                "series_id": None,
                "provider": "fred",
            }
        ),
        (
            {
                "query": None,
                "search_type": "full_text",
                "release_id": None,
                "offset": None,
                "limit": None,
                "order_by": "observation_end",
                "sort_order": "desc",
                "filter_variable": None,
                "filter_value": None,
                "tag_names": None,
                "exclude_tag_names": None,
                "series_id": "NYICLAIMS",
                "provider": "fred",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_fred_search(params, obb):
    """Test economy fred search."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.fred_search(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "symbol": "SP500",
                "start_date": None,
                "end_date": None,
                "limit": 10000,
                "frequency": "q",
                "aggregation_method": "eop",
                "transform": "chg",
                "provider": "fred",
            }
        ),
        (
            {
                "symbol": "FEDFUNDS",
                "start_date": None,
                "end_date": None,
                "limit": 10000,
                "all_pages": True,
                "provider": "intrinio",
                "sleep": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_fred_series(params, obb):
    """Test economy fred series."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.fred_series(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        ({"start_date": "2023-01-01", "end_date": "2023-06-06", "adjusted": True}),
        (
            {
                "provider": "federal_reserve",
                "start_date": "2023-01-01",
                "end_date": "2023-06-06",
                "adjusted": True,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_money_measures(params, obb):
    """Test economy money measures."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.money_measures(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states",
                "sex": "total",
                "frequency": "monthly",
                "age": "total",
                "seasonal_adjustment": True,
                "provider": "oecd",
                "start_date": "2023-01-01",
                "end_date": "2023-06-06",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_unemployment(params, obb):
    """Test economy unemployment."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.unemployment(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states",
                "adjustment": "amplitude",
                "growth_rate": False,
                "provider": "oecd",
                "start_date": "2023-01-01",
                "end_date": "2023-06-06",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_composite_leading_indicator(params, obb):
    """Test economy composite leading indicator."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.composite_leading_indicator(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "fred",
                "symbol": "156241",
                "is_series_group": True,
                "start_date": "2000-01-01",
                "end_date": None,
                "frequency": "w",
                "units": "Number",
                "region_type": "state",
                "season": "nsa",
                "aggregation_method": "eop",
                "transform": "ch1",
                "limit": None,
            }
        ),
        (
            {
                "provider": "fred",
                "symbol": "CAICLAIMS",
                "is_series_group": False,
                "start_date": "1990-01-01",
                "end_date": "2010-01-01",
                "frequency": None,
                "units": None,
                "region_type": None,
                "season": None,
                "aggregation_method": "avg",
                "transform": "chg",
                "limit": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_fred_regional(params, obb):
    """Test economy fred regional."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.fred_regional(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "econdb",
                "country": "us,uk,jp",
                "latest": True,
                "use_cache": False,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_country_profile(params, obb):
    """Test economy country profile."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.country_profile(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        ({"provider": "econdb", "use_cache": False}),
        (
            {
                "provider": "imf",
                "query": "gold+volume",
                "dataflows": None,
                "keywords": None,
                "symbol": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_available_indicators(params, obb):
    """Test economy available indicators."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.available_indicators(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "econdb",
                "country": "us,uk,jp",
                "symbol": "GDP,GDEBT",
                "transform": None,
                "start_date": "2022-01-01",
                "end_date": "2024-01-01",
                "use_cache": False,
                "frequency": None,
            }
        ),
        (
            {
                "provider": "econdb",
                "country": None,
                "symbol": "MAIN",
                "transform": None,
                "start_date": "2022-01-01",
                "end_date": "2024-01-01",
                "use_cache": False,
                "frequency": "quarter",
            }
        ),
        (
            {
                "provider": "imf",
                "country": "*",
                "symbol": "IL::RGV_REVS",
                "start_date": "2025-09-30",
                "end_date": None,
                "frequency": "month",
                "transform": None,
                "dimension_values": None,
                "limit": 1,
                "pivot": False,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_indicators(params, obb):
    """Test economy indicators."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.indicators(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "date": None,
                "provider": "federal_reserve",
                "holding_type": "all_treasury",
                "summary": False,
                "monthly": False,
                "cusip": None,
                "wam": False,
            }
        ),
        (
            {
                "date": None,
                "provider": "federal_reserve",
                "holding_type": "all_agency",
                "summary": False,
                "monthly": False,
                "cusip": None,
                "wam": True,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_central_bank_holdings(params, obb):
    """Test economy central bank holdings."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.central_bank_holdings(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states,united_kingdom",
                "frequency": "monthly",
                "provider": "oecd",
                "start_date": "2022-01-01",
                "end_date": "2024-04-01",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_share_price_index(params, obb):
    """Test economy share price index."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.share_price_index(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states,united_kingdom",
                "frequency": "quarter",
                "provider": "oecd",
                "start_date": "2022-01-01",
                "end_date": "2024-04-01",
                "transform": "index",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_house_price_index(params, obb):
    """Test economy house price index."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.house_price_index(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states",
                "frequency": "monthly",
                "provider": "oecd",
                "start_date": "2023-01-01",
                "end_date": "2023-06-06",
                "duration": "long",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_interest_rates(params, obb):
    """Test economy country interest rates endpoint."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.interest_rates(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "country": "united_states",
                "item": "meats",
                "region": "all_city",
                "frequency": "annual",
                "provider": "fred",
                "start_date": "2022-01-01",
                "end_date": "2024-04-01",
                "transform": "pc1",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_retail_prices(params, obb):
    """Test economy retail prices."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.retail_prices(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "frequency": None,
                "provider": "fred",
                "start_date": "2022-01-01",
                "end_date": "2024-04-01",
                "transform": None,
                "aggregation_method": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_survey_university_of_michigan(params, obb):
    """Test the economy survey university_of_michigan endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.university_of_michigan(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "category": "auto",
                "provider": "fred",
                "start_date": "2022-01-01",
                "end_date": "2024-04-01",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_survey_sloos(params, obb):
    """Test the economy survey sloos endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.sloos(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "fred",
                "start_date": "2024-01-01",
                "end_date": "2024-04-01",
                "transform": None,
                "aggregation_method": None,
                "frequency": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_survey_economic_conditions_chicago(params, obb):
    """Test the economy survey economic conditions chicago endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.economic_conditions_chicago(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "fred",
                "topic": "new_orders",
                "start_date": "2024-01-01",
                "end_date": "2024-04-01",
                "transform": None,
                "aggregation_method": None,
                "frequency": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_survey_manufacturing_outlook_texas(params, obb):
    """Test the economy survey manufacturing outlook texas endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.manufacturing_outlook_texas(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "fred",
                "topic": "new_orders",
                "start_date": "2024-01-01",
                "end_date": "2024-04-01",
                "transform": None,
                "aggregation_method": None,
                "frequency": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_survey_manufacturing_outlook_ny(params, obb):
    """Test the economy survey manufacturing outlook ny endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.manufacturing_outlook_ny(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "federal_reserve",
                "start_date": "2024-01-01",
                "end_date": "2024-04-01",
                "category": "cmbs",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_primary_dealer_positioning(params, obb):
    """Test the economy primary dealer positioning endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.primary_dealer_positioning(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "fred",
                "date": "2024-06-01,2023-06-01",
                "category": "avg_earnings_hourly",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_survey_nonfarm_payrolls(params, obb):
    """Test the economy survey nonfarm payrolls endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.nonfarm_payrolls(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "fred",
                "date": "2024-05-01,2024-04-01,2023-05-01",
                "category": "pce_price_index",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_pce(params, obb):
    """Test the economy pce endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.pce(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "fred",
                "date": None,
                "release_id": "14",
                "element_id": "7930",
            }
        ),
        (
            {
                "provider": "fred",
                "date": None,
                "release_id": "14",
                "element_id": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_fred_release_table(params, obb):
    """Test the economy fred release table endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.fred_release_table(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "bls",
                "query": "gasoline;seattle;average price",
                "category": "cpi",
                "include_extras": False,
                "include_code_map": False,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_survey_bls_search(params, obb):
    """Test the economy survey bls search endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.bls_search(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "bls",
                "symbol": "APUS49D74714,APUS49D74715,APUS49D74716",
                "start_date": "2024-01-01",
                "end_date": "2024-07-01",
                "aspects": False,
                "calculations": True,
                "annual_average": True,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_survey_bls_series(params, obb):
    """Test the economy survey bls series endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.bls_series(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "econdb",
                "country": "IN,CN",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_export_destinations(params, obb):
    """Test the economy export destinations endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.export_destinations(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "federal_reserve",
                "start_date": None,
                "end_date": None,
                "asset_class": "mbs",
                "unit": "value",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_primary_dealer_fails(params, obb):
    """Test the economy primary dealer fails endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.primary_dealer_fails(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "imf",
                "country": "us",
                "counterpart": "world,eu",
                "frequency": "annual",
                "direction": "exports",
                "start_date": "2020-01-01",
                "end_date": "2023-01-01",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_direction_of_trade(params, obb):
    """Test the economy direction of trade endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.direction_of_trade(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "federal_reserve",
                "year": 2022,
                "document_type": "minutes",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_fomc_documents(params, obb):
    """Test the economy fomc documents endpoint"""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.fomc_documents(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        {
            "provider": "imf",
            "port_code": None,
            "country": None,
            "continent": None,
            "limit": None,
        }
    ],
)
@pytest.mark.integration
def test_economy_shipping_port_info(params, obb):
    """Test economy shipping port info."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.shipping_port_info(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "imf",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_shipping_chokepoint_info(params, obb):
    """Test economy shipping chokepoint info."""
    result = obb.economy.shipping.chokepoint_info(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "imf",
                "chokepoint": "chokepoint1",
                "start_date": "2023-01-01",
                "end_date": "2023-01-31",
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_shipping_chokepoint_volume(params, obb):
    """Test economy shipping chokepoint volume."""
    result = obb.economy.shipping.chokepoint_volume(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        (
            {
                "provider": "imf",
                "port_code": "port1201",
                "start_date": "2023-01-01",
                "end_date": "2023-01-31",
                "country": None,
            }
        ),
        (
            {
                "provider": "econdb",
                "start_date": None,
                "end_date": None,
            }
        ),
    ],
)
@pytest.mark.integration
def test_economy_shipping_port_volume(params, obb):
    """Test economy shipping chokepoint volume."""
    result = obb.economy.shipping.chokepoint_volume(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.skip(reason="Endpoint not available to Python SDK.")
def test_economy_fomc_documents_download(obb):
    """Test the economy fomc documents download endpoint."""
    params = {
        "url": "https://www.federalreserve.gov/monetarypolicy/files/BeigeBook_20230118.pdf"
    }
    result = obb.economy.fomc_documents_download(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        {
            "provider": "federal_reserve",
            "frequency": "summary",
            "start_date": None,
            "end_date": None,
        }
    ],
)
@pytest.mark.integration
def test_economy_total_factor_productivity(params, obb):
    """Test economy total factor productivity."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.total_factor_productivity(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0


@pytest.mark.parametrize(
    "params",
    [
        {
            "provider": "federal_reserve",
            "start_date": "2020-01-01",
            "end_date": "2024-12-31",
        }
    ],
)
@pytest.mark.integration
def test_economy_survey_inflation_expectations(params, obb):
    """Test economy survey inflation expectations."""
    params = {p: v for p, v in params.items() if v}

    result = obb.economy.survey.inflation_expectations(**params)
    assert result
    assert isinstance(result, OBBject)
    assert len(result.results) > 0
