Files
OpenBB/tests/test_fa/test_fa_excel.py
colin99d ce457314a0 Improved Discount Rate Calculation for DCF (#697)
* Added tests ReadMe

* Added warning

* Moved testing docs

* Fixed formatting

* Added a CF tie-out warning and handling for negative firm value

* Added treasury API

* Updated t-bill from API

* Finished better risk understanding

* Finished better risk understanding

* Requested changes

* Updated

* Updated filename

Co-authored-by: Colin Delahunty <colindelahunty@colins-air.attlocal.net>
Co-authored-by: Colin Delahunty <colindelahunty@Colins-MacBook-Air.local>
2021-08-22 01:23:44 +01:00

30 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",
record_mode="new_episodes",
)
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)