def test_shaping_reward_is_shaping(self):
base_reward = (1, 2)
shaping_reward = (3, 4)
reward = Reward(base_reward, shaping_reward)
self.assertTrue(reward.is_shaping())
def test_base_reward_is_not_shaping(self):
base_reward = (1, 2)
shaping_reward = ()
reward = Reward(base_reward, shaping_reward)
self.assertFalse(reward.is_shaping())
