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())