def test_do_calculate_reward_if_yaw(): env = NavigationTrack() env._ranges = np.array([2, 1, 10, 4, 5]) reward1 = env._do_calculate_reward(env._YAW_RIGHT) reward2 = env._do_calculate_reward(env._YAW_LEFT) assert reward1 == reward2 == -0.5
def test_do_calculate_reward_if_forward(): env = NavigationTrack() env._ranges = np.array([2, 1, 10, 4, 5]) reward = env._do_calculate_reward(env._FORWARD) assert reward == 5
def test_do_calculate_reward_if_done(): env = NavigationTrack() env._ranges = np.array([0.5, 1, 0.01, 4, 5]) reward = env._do_calculate_reward(env._FORWARD) assert reward == -200