fix test

hello-robot · Aug 21, 2024 · 6a2094e · 6a2094e
1 parent b5c9b8a
commit 6a2094e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/body/test/robot_tests/test_robot_home_stow.py b/body/test/robot_tests/test_robot_home_stow.py
@@ -96,8 +96,8 @@ def test_robot_home(self):
 
         if self.robot.params['tool'] == 'eoa_wrist_dw3_tool_sg3':
             home_poses['wrist_yaw'] = 0
-            home_poses['wrist_roll'] = self.robot.params['stow']['wrist_roll']
-            home_poses['wrist_pitch'] = self.robot.params['stow']['wrist_pitch']
+            home_poses['wrist_roll'] = self.robot.end_of_arm.params['stow']['wrist_roll']
+            home_poses['wrist_pitch'] = self.robot.end_of_arm.params['stow']['wrist_pitch']
             home_poses['stretch_gripper'] = 0
 
         if self.robot.params['tool'] == 'tool_stretch_gripper':
@@ -106,8 +106,8 @@ def test_robot_home(self):
 
         if self.robot.params['tool'] == 'eoa_wrist_dw3_tool_nil':
             home_poses['wrist_yaw'] = 0
-            home_poses['wrist_roll'] = self.robot.params['stow']['wrist_roll']
-            home_poses['wrist_pitch'] = self.robot.params['stow']['wrist_pitch']
+            home_poses['wrist_roll'] = self.robot.end_of_arm.params['stow']['wrist_roll']
+            home_poses['wrist_pitch'] = self.robot.end_of_arm.params['stow']['wrist_pitch']
 
         self.check_joint_poses(home_poses, 'Homing')