From f88da309f1326841c38cffcd73a2a6cfb89e7620 Mon Sep 17 00:00:00 2001 From: Artem Veremey Date: Wed, 3 Mar 2021 21:40:49 -0800 Subject: [PATCH] Fix a big for Yahoo FA when a ticker has no sustainability info --- .../fundamental_analysis/yahoo_finance_api.py | 4 ++++ .../tests/test_fa_yahoo_finance_api.py | 12 +++++++++++- 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/gamestonk_terminal/fundamental_analysis/yahoo_finance_api.py b/gamestonk_terminal/fundamental_analysis/yahoo_finance_api.py index f579ca6991b..91a2837cbf5 100644 --- a/gamestonk_terminal/fundamental_analysis/yahoo_finance_api.py +++ b/gamestonk_terminal/fundamental_analysis/yahoo_finance_api.py @@ -186,6 +186,10 @@ def sustainability(l_args, s_ticker): df_sustainability = stock.sustainability + if not df_sustainability: + print(f"No sustainability information in Yahoo for {s_ticker}") + return + df_sustainability.index = [ "".join( " " + char if char.isupper() else char.strip() for char in idx diff --git a/gamestonk_terminal/tests/test_fa_yahoo_finance_api.py b/gamestonk_terminal/tests/test_fa_yahoo_finance_api.py index 2af4947e819..4cdb7411c9c 100755 --- a/gamestonk_terminal/tests/test_fa_yahoo_finance_api.py +++ b/gamestonk_terminal/tests/test_fa_yahoo_finance_api.py @@ -5,9 +5,19 @@ import unittest # pylint: disable=unused-import import pytest -from gamestonk_terminal.fundamental_analysis.yahoo_finance_api import info +from gamestonk_terminal.fundamental_analysis.yahoo_finance_api import ( + info, + sustainability, + calendar_earnings, +) class TestFaYahooFinanceApi(unittest.TestCase): def test_info(self): info([], "PLTR") + + def test_sustainability(self): + sustainability([], "PLTR") + + def test_calendar_earnings(self): + calendar_earnings([], "PLTR")