Merge pull request #37 from joshyattridge/unit_testing

Added unit testing
joshyattridge · May 14, 2024 · 58bb145 · 58bb145
2 parents da9ca70 + 0c8c0ae
commit 58bb145
Show file tree

Hide file tree

Showing 11 changed files with 219,937 additions and 2 deletions.
diff --git a/smartmoneyconcepts/smc.py b/smartmoneyconcepts/smc.py
@@ -475,7 +475,7 @@ def ob(
                     )
                     percentage[obIndex] = (
                         np.min([highVolume[obIndex], lowVolume[obIndex]], axis=0)
-                        / np.max([highVolume[obIndex], lowVolume[obIndex]], axis=0)
+                        / np.max([highVolume[obIndex], lowVolume[obIndex]], axis=0) if np.max([highVolume[obIndex], lowVolume[obIndex]], axis=0) != 0 else 1
                     ) * 100.0
 
         for i in range(len(ohlc)):
@@ -547,7 +547,7 @@ def ob(
                     highVolume[obIndex] = ohlc["volume"].iloc[close_index - 2]
                     percentage[obIndex] = (
                         np.min([highVolume[obIndex], lowVolume[obIndex]], axis=0)
-                        / np.max([highVolume[obIndex], lowVolume[obIndex]], axis=0)
+                        / np.max([highVolume[obIndex], lowVolume[obIndex]], axis=0) if np.max([highVolume[obIndex], lowVolume[obIndex]], axis=0) != 0 else 1
                     ) * 100.0
 
         ob = np.where(ob != 0, ob, np.nan)

diff --git a/tests/test_data/EURUSD_15M.csv b/tests/test_data/EURUSD_15M.csv