Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_rsi():
"""test TA.RSI"""
rsi = TA.RSI(ohlc)
assert isinstance(rsi, series.Series)
assert -100 < rsi.values[-1] < 100
def calc_ta(self):
ta = [
TA.ER(self.ohlc),
TA.PPO(self.ohlc)["HISTO"],
TA.STOCHRSI(self.ohlc),
TA.ADX(self.ohlc),
TA.RSI(self.ohlc),
TA.COPP(self.ohlc),
TA.CCI(self.ohlc),
TA.CHAIKIN(self.ohlc),
TA.FISH(self.ohlc)
]
ta = np.array(ta).transpose()
ta[np.isnan(ta)] = 0
ta[np.isinf(ta)] = 0
# scale them to the same range
self.scaler = StandardScaler()
self.scaler.fit(ta)
self.ta = self.scaler.transform(ta)