def ignore_test_rating_below_zero(self):
pm = VisionModule()
pm.start(self.data)
self.data_mock["CameraFrameVersion"] = 1
self.data_mock["CameraPose"] = test_utils.get_init_pose()
ball_info = (-1, (1, 2, 3))
pm.extract_ball_infos(self.data_mock, ball_info)
self.assertFalse(self.data_mock["BallFound"])
def test_save_color_config(self):
pm=VisionModule()
pm.start(self.data)
name = "all"
path2res = find_resource("vision-color-config/auto-color-config")
system("rm %s/%s.png" % (path2res, name))
pm.save_color_config(name)
self.assertTrue(path.isfile("%s/%s.png" % (path2res, name)))
def test_extract_goal_info(self):
pm = VisionModule()
pm.start(self.data)
p = test_utils.get_init_pose()
self.data_mock["Pose"] = p
pm.transformer.update_pose(self.data_mock["Pose"])
self.data_mock["CameraFrameVersion"] = 1
goal_info = (((0,0,0),(0.0,0.0,0)), None)
pm.extract_goal_infos(self.data_mock, goal_info)
self.assertTrue(self.data_mock[DATA_KEY_GOAL_FOUND])
self.assertTrue(len(self.data_mock[DATA_KEY_GOAL_INFO]) == 2)
def ignore_test_new_camera_frame_is_really_true(self):
# Create Vision Module
pm = VisionModule()
pm.start(self.data)
self.data_mock["CameraFrameVersion"] = 1
result = pm.check_is_new_frame(self.data_mock)
self.assertTrue(result)
self.data_mock["CameraFrameVersion"] = 2
result = pm.check_is_new_frame(self.data_mock)
self.assertTrue(result)
self.data_mock["CameraFrameVersion"] = 3
result = pm.check_is_new_frame(self.data_mock)
self.assertTrue(result)
self.data_mock["CameraFrameVersion"] = 3
result = pm.check_is_new_frame(self.data_mock)
self.assertFalse(result)
self.data_mock["CameraFrameVersion"] = 0
result = pm.check_is_new_frame(self.data_mock)
self.assertFalse(result)
self.data_mock["CameraFrameVersion"] = 4
result = pm.check_is_new_frame(self.data_mock)
self.assertTrue(result)
