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_adx():
"""test TA.ADX"""
adx = TA.ADX(ohlc)
assert isinstance(adx, series.Series)
assert adx.values[-1] == 54.781005617031234
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)