def test_donchian():
candles = np.array(mama_candles)
single = ta.donchian(candles, period=20)
seq = ta.donchian(candles, period=20, sequential=True)
assert type(single).__name__ == 'DonchianChannel'
assert round(single.upperband, 2) == 277.20
assert round(single.middleband, 2) == 189.20
assert round(single.lowerband, 2) == 101.20
assert seq.middleband[-1] == single.middleband
assert len(seq.upperband) == len(candles)
assert len(seq.middleband) == len(candles)
assert len(seq.lowerband) == len(candles)
def exit_donchian(self):
return donchian(self.candles, self.vars["exit_dc_period"])
def donchain(self):
return ta.donchian(self.candles[:-1])
def donchian(self):
# Previous Donchian Channels with default parameters
return ta.donchian(self.candles[:-1])
