def test_rewards(self):
env = Environment()
env.reset()
state, reward, done = env.step({"buy": 3, "sell": 0})
sell_price = env._get_state(False)[0].iloc[-1]["close"]
state, reward, done = env.step({"buy": 0, "sell": 3})
self.assertEqual(reward, sell_price * 3,
"Rewards are Estimated Correctly")
