fix zero divide warning

msaltnet · May 6, 2024 · 02fb5c0 · 02fb5c0
1 parent 087e910
commit 02fb5c0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/smtm/analyzer.py b/smtm/analyzer.py
@@ -348,6 +348,8 @@ def make_rsi(prices, count=14):
         seed = deltas[:count]
         up_avg = seed[seed >= 0].sum() / count
         down_avg = -seed[seed < 0].sum() / count
+        if down_avg == 0:
+            return None
         r_strength = up_avg / down_avg
         rsi = np.zeros_like(prices)
         rsi[: count + 1] = 100.0 - 100.0 / (1.0 + r_strength)
@@ -365,6 +367,8 @@ def make_rsi(prices, count=14):
             up_avg = (up_avg * (count - 1) + upval) / count
             down_avg = (down_avg * (count - 1) + downval) / count
 
+            if down_avg == 0:
+                return None
             r_strength = up_avg / down_avg
             rsi[i] = 100.0 - 100.0 / (1.0 + r_strength)
 

diff --git a/tests/unit_tests/analyzer_test.py b/tests/unit_tests/analyzer_test.py
@@ -1142,7 +1142,7 @@ def test_create_report_draw_correct_graph_when_rsi_enabled(
         analyzer.update_asset_info = MagicMock()
         filename = "apple"
 
-        self.fill_test_data_for_report(analyzer)
+        self.fill_test_data_for_report_10(analyzer)
         analyzer.create_report(filename)
 
         self.assertEqual(