From cf029cfae2b1dc6807715919fc8b9e35e1d2c42a Mon Sep 17 00:00:00 2001 From: Artem Veremey Date: Fri, 2 Apr 2021 21:51:06 -0700 Subject: [PATCH] Adding Fred unit test --- tests/test_fred_fred_view.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 tests/test_fred_fred_view.py diff --git a/tests/test_fred_fred_view.py b/tests/test_fred_fred_view.py new file mode 100644 index 00000000000..96b97919c1b --- /dev/null +++ b/tests/test_fred_fred_view.py @@ -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")