def test_SMA20(self): return f = open(TEST_SMA_INPUT_FILE, "r") sys.stdin = StringIO(f.read()) f.close() b = Bot() b.run() selected_candles = b.get_selected_candles(CurrencyEnum.USDT_ETH)[:20] closing_prices = [] for candle in selected_candles: closing_prices.append(candle.close) mean = SMA.SMA(closing_prices) r = pd.Series(closing_prices) panda_means = r.rolling(window=20).mean() panda_mean = panda_means.tolist()[19] self.assertTrue(mean == panda_mean)