Minor updates

JerBouma · Jan 9, 2024 · 0cc1496 · 0cc1496
1 parent 588f50a
commit 0cc1496
Show file tree

Hide file tree

Showing 24 changed files with 1,231 additions and 7 deletions.
diff --git a/financetoolkit/historical_model.py b/financetoolkit/historical_model.py
@@ -1,6 +1,7 @@
 """Historical Module"""
 __docformat__ = "google"
 
+import contextlib
 import threading
 import time
 from datetime import datetime, timedelta
@@ -557,7 +558,9 @@ def enrich_historical_data(
     historical_data["Cumulative Return"] = 1
 
     adjusted_return = historical_data.loc[start:end, "Return"].copy()
-    adjusted_return.iloc[0] = 0
+
+    with contextlib.suppress(IndexError):
+        adjusted_return.iloc[0] = 0
 
     historical_data["Cumulative Return"] = pd.Series(np.nan).astype(float)
 

diff --git a/financetoolkit/toolkit_controller.py b/financetoolkit/toolkit_controller.py
@@ -497,12 +497,6 @@ def ratios(self) -> Ratios:
         if self._cash_flow_statement.empty:
             empty_data.append("Cash Flow Statement")
 
-        empty_data_iterator = (
-            tqdm(empty_data, desc="Obtaining financial statements")
-            if ENABLE_TQDM & self._progress_bar
-            else empty_data
-        )
-
         if empty_data:
             empty_data_iterator = (
                 tqdm(empty_data, desc="Obtaining financial statements")

diff --git a/tests/models/csv/test_growth_model/test_get_present_value_of_growth_opportunities.csv b/tests/models/csv/test_growth_model/test_get_present_value_of_growth_opportunities.csv
@@ -0,0 +1,4 @@
+,0
+0,0.6666666666666666
+1,1.6
+2,2.7
diff --git a/tests/models/csv/test_models_controller/test_get_present_value_of_growth_opportunities.csv b/tests/models/csv/test_models_controller/test_get_present_value_of_growth_opportunities.csv
@@ -0,0 +1,4 @@
+,AAPL,MSFT
+2020,8318.9299,57852.0784
+2021,160.2573,306.022
+2022,155.4936,279.3459