def test_bad_trust_at_start(self, part: AngleRoadPart):
assert part.run(road_ellipse=Ellipse((70, 71), (100, 200), 135.0, trust=0.1)) == 0.0, \
'Angle should be ignored when no previous value and bad trust value'
assert part.run(road_ellipse=Ellipse((70, 71), (100, 200), 135.0, trust=0.1)) == 0.0, \
'No previous angle value'
assert part.run(road_ellipse=Ellipse((70, 71), (100, 200), 135.0, trust=0.5)) == -0.5, \
'At startup, use average between angle and straight ahead'
def test_straight_ahead(self, part: AngleRoadPart):
assert 0 < part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 89.0, 1.0)) < 0.1
assert -0.1 < part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 91.0, 1.0)) < 0.0
assert 0 < part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 269.0, 1.0)) < 0.1
assert -0.1 < part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 271.0, 1.0)) < 0.0
def test_trust_not_perfect(self, part: AngleRoadPart):
# Record previous value
assert part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 45.0, trust=1.0)) == 1.0
assert part.run(road_ellipse=Ellipse((70, 71), (100, 200), 135.0, trust=0.5)) == 0.25, \
'At startup, use weighted average between current and previous angles'
def test_bad_trust(self, part: AngleRoadPart):
assert 0 < part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 89.0, trust=1.0)) < 0.1
assert 0 < part.run(road_ellipse=Ellipse((70, 71), (100, 200), 135.0, trust=0.1)) < 0.1, \
'Previous value should be used when trust < 0.5'
def test_turn_left(self, part: AngleRoadPart):
assert part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 135.0, 1.0)) == -1.0
assert part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 180.0, 1.0)) == -1.0
def test_turn_right(self, part: AngleRoadPart):
assert part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 45.0, 1.0)) == 1.0
assert part.run(
road_ellipse=Ellipse((70, 71), (100, 200), 0.1, 1.0)) == 1.0