adding rsi tests

bukosabino · Nov 28, 2019 · 9142767 · 9142767
1 parent fda6f23
commit 9142767
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ta/momentum.py b/ta/momentum.py
@@ -39,7 +39,7 @@ def _run(self):
         emaup = up.ewm(alpha=1/self._n, min_periods=0, adjust=False).mean()
         emadn = dn.ewm(alpha=1/self._n, min_periods=0, adjust=False).mean()
         rs = emaup / emadn
-        self._rsi = np.where(emadn == 0, 100, 100-(100/(1+rs)))
+        self._rsi = pd.Series(np.where(emadn == 0, 100, 100-(100/(1+rs))), index=self._close.index)
 
     def rsi(self) -> pd.Series:
         """Relative Strength Index (RSI)