mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-02 15:32:04 +00:00
ec07686aed
* Allow to quit when DEFAULT_CONTEXT selected * terminal imports inside context selection * rework economy/events command * Refactor fear greed CNN command * refactor WSJ in economy * import fred * refactor fred commands in economy * refactor disc commands into economy * refactor disc commands into economy 2 * fix fred test * fix fred test * av pylint * update quit from direct context * fix av lint * Add industry simply wallst to economy * improve events display * improv after review * add mypy types * tabulate raw series * remove simply wallst
42 lines
1018 B
Python
42 lines
1018 B
Python
""" econ/fred_view.py tests """
|
|
import unittest
|
|
from unittest import mock
|
|
from io import StringIO
|
|
import pandas as pd
|
|
|
|
# pylint: disable=unused-import
|
|
from gamestonk_terminal.economy.fred_view import display_series # noqa: F401
|
|
|
|
fred_data_mock = """
|
|
For 'gdp', series IDs found: GDP, GDPC1, M2V, GFDEGDQ188S, PAYEMS.
|
|
|
|
GDP
|
|
---
|
|
Date
|
|
2019-01-01 21001.591
|
|
2019-04-01 21289.268
|
|
2019-07-01 21505.012
|
|
2019-10-01 21694.458
|
|
2020-01-01 21481.367
|
|
2020-04-01 19477.444
|
|
2020-07-01 21138.574
|
|
2020-10-01 21477.597
|
|
2021-01-01 22038.226
|
|
2021-04-01 22722.581
|
|
"""
|
|
|
|
|
|
class TestFredFredView(unittest.TestCase):
|
|
@mock.patch("gamestonk_terminal.economy.fred_model.get_series_data")
|
|
def test_display_fred(self, mock_get_series):
|
|
fred_data = pd.read_csv(StringIO(fred_data_mock), header=0, index_col=0)
|
|
|
|
mock_get_series.return_value = fred_data
|
|
|
|
display_series(
|
|
series="gdp",
|
|
start_date="2019-01-01",
|
|
raw=True,
|
|
export="",
|
|
)
|