diff --git a/turn_test.py b/turn_test.py index 212a1c6..0fce3f7 100644 --- a/turn_test.py +++ b/turn_test.py @@ -14,6 +14,13 @@ def main(): else: power = int(power) + # Gets the offset for the robot + offset = input("Offset (4):") + if offset == "": + offset = 64 + else: + offset = int(offset) + # Calculates sleep times for the robot turn_t = DRIVE_A / power # 10 degrees turn at given power @@ -22,8 +29,14 @@ def main(): arlo.go_diff(power, power + RIGHT_WHEEL_OFFSET, 1, 0) sleep(turn_t * 36) + + arlo.stop() + sleep(1) + + arlo.go_diff(power, power + RIGHT_WHEEL_OFFSET, 0, 1) + sleep(turn_t * 36) + arlo.stop() if __name__ == "__main__": - main()