mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 21:12:16 +00:00
Adding Fred unit test
This commit is contained in:
30
tests/test_fred_fred_view.py
Normal file
30
tests/test_fred_fred_view.py
Normal file
@@ -0,0 +1,30 @@
|
||||
""" fred/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.fred.fred_view import get_fred_data # noqa: F401
|
||||
|
||||
fred_data_mock = """
|
||||
,GDP
|
||||
2019-01-01,21115.309
|
||||
2019-04-01,21329.877
|
||||
2019-07-01,21540.325
|
||||
2019-10-01,21747.394
|
||||
2020-01-01,21561.139
|
||||
2020-04-01,19520.114
|
||||
2020-07-01,21170.252
|
||||
2020-10-01,21494.731
|
||||
"""
|
||||
|
||||
|
||||
class TestFredFredView(unittest.TestCase):
|
||||
@mock.patch("gamestonk_terminal.fred.fred_view.Fred.get_series")
|
||||
def test_get_fred_data(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
|
||||
|
||||
get_fred_data(["--noplot"], "gdp")
|
||||
Reference in New Issue
Block a user