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)