Files
OpenBB/tests/test_fa/test_fa_excel.py
2021-08-16 07:59:52 -04:00

29 lines
1.1 KiB
Python

""" fundamental_analysis/dcf_view.py tests """
from unittest import TestCase
import vcr
from gamestonk_terminal.stocks.fundamental_analysis import dcf_model
from gamestonk_terminal.stocks.fundamental_analysis.dcf_view import CreateExcelFA
class TestExcelClass(TestCase):
@vcr.use_cassette(
"tests/cassettes/test_fa/test_fa_excel/test_covers_all_tickers.yaml"
)
def test_covers_all_tickers(self):
for ticker in dcf_model.tickers:
excel = CreateExcelFA(ticker, False)
df_is = excel.get_data("IS", 1, True)
items_is = dcf_model.non_gaap_is + dcf_model.gaap_is
for item in df_is.index:
self.assertIn(item, items_is)
df_bs = excel.get_data("BS", 1, True)
items_bs = dcf_model.non_gaap_bs + dcf_model.gaap_bs
for item in df_bs.index:
self.assertIn(item, items_bs)
df_cf = excel.get_data("CF", 1, True)
items_cf = dcf_model.non_gaap_cf + dcf_model.gaap_cf
for item in df_cf.index:
self.assertIn(item, items_cf)