Fix robotDrive tests

ryannazaretian · Feb 17, 2016 · 95ad26b · 95ad26b
1 parent 6b6adb8
commit 95ad26b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/wpilib/tests/test_robotdrive.py b/wpilib/tests/test_robotdrive.py
@@ -485,13 +485,13 @@ def test_getDescription(drive_lr):
 
 def test_stopMotor(drive_lr):
     drive_lr.stopMotor()
-    drive_lr.rearLeftMotor.set.assert_called_once_with(0.0)
-    drive_lr.rearRightMotor.set.assert_called_once_with(0.0)
+    drive_lr.rearLeftMotor.stopMotor.assert_called_once_with()
+    drive_lr.rearRightMotor.stopMotor.assert_called_once_with()
 
 def test_stopMotor_4(drive4):
     drive4.stopMotor()
-    drive4.frontLeftMotor.set.assert_called_once_with(0.0)
-    drive4.frontRightMotor.set.assert_called_once_with(0.0)
-    drive4.rearLeftMotor.set.assert_called_once_with(0.0)
-    drive4.rearRightMotor.set.assert_called_once_with(0.0)
+    drive4.frontLeftMotor.stopMotor.assert_called_once_with()
+    drive4.frontRightMotor.stopMotor.assert_called_once_with()
+    drive4.rearLeftMotor.stopMotor.assert_called_once_with()
+    drive4.rearRightMotor.stopMotor.assert_called_once_with()