Python NavigationTrack._do_calculate_reward Exemples

Exemple #1

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

Exemple #2

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

Exemple #3

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