summaryrefslogtreecommitdiffstats
path: root/openbb_platform/providers/fred/tests/test_fred_fetchers.py
diff options
context:
space:
mode:
Diffstat (limited to 'openbb_platform/providers/fred/tests/test_fred_fetchers.py')
-rw-r--r--openbb_platform/providers/fred/tests/test_fred_fetchers.py21
1 files changed, 17 insertions, 4 deletions
diff --git a/openbb_platform/providers/fred/tests/test_fred_fetchers.py b/openbb_platform/providers/fred/tests/test_fred_fetchers.py
index 0529d88ad84..45a04d35d35 100644
--- a/openbb_platform/providers/fred/tests/test_fred_fetchers.py
+++ b/openbb_platform/providers/fred/tests/test_fred_fetchers.py
@@ -29,7 +29,10 @@ from openbb_fred.models.sonia_rates import FREDSONIAFetcher
from openbb_fred.models.spot import FREDSpotRateFetcher
from openbb_fred.models.tbffr import FREDSelectedTreasuryBillFetcher
from openbb_fred.models.tmc import FREDTreasuryConstantMaturityFetcher
-from openbb_fred.models.us_yield_curve import FREDYieldCurveFetcher
+from openbb_fred.models.us_yield_curve import (
+ FREDYieldCurveFetcher as FREDUSYieldCurveFetcher,
+)
+from openbb_fred.models.yield_curve import FREDYieldCurveFetcher
test_credentials = UserService().default_user_settings.credentials.model_dump(
mode="json"
@@ -58,11 +61,11 @@ def test_fredcpi_fetcher(credentials=test_credentials):
@pytest.mark.record_http
-def test_fred_yield_curve_fetcher(credentials=test_credentials):
- """Test FREDYieldCurveFetcher."""
+def test_fred_us_yield_curve_fetcher(credentials=test_credentials):
+ """Test FREDUSYieldCurveFetcher."""
params = {"date": datetime.date(2023, 12, 1)}
- fetcher = FREDYieldCurveFetcher()
+ fetcher = FREDUSYieldCurveFetcher()
result = fetcher.test(params, credentials)
assert result is None
@@ -302,3 +305,13 @@ def test_fred_regional_fetcher(credentials=test_credentials):
fetcher = FredRegionalDataFetcher()
result = fetcher.test(params, credentials)
assert result is None
+
+
+@pytest.mark.record_http
+def test_fred_yield_curve_fetcher(credentials=test_credentials):
+ """Test FREDYieldCurveFetcher."""
+ params = {"date": "2024-05-14,2023-05-14,2022-03-16,2021-05-14,2020-05-14"}
+
+ fetcher = FREDYieldCurveFetcher()
+ result = fetcher.test(params, credentials)
+ assert result is None