def test_set_region_of_interest_rel_points_should_create_expected_roi( self, mock_camera): cam = mock_camera.return_value cam.shape = [300, 100] api = ScannerAPI() expected_roi = ROI(0.0, 0.0, 0.5, 0.9) api.set_region_of_interest_from_rel_points(0, 0, 0.5, 0.9) self.assertROIEquals(expected_roi, api.roi)
def __init__(self): self.camera = Camera() self.camera.start() self._default_roi = ROI(0.0, 0.0, 1.0, 1.0) self._default_encoder = Encoder((0.2, 0.2), 382, 100, 20, 200) self._default_laser_detector = LaserDetector2(225, 'red') self.encoder = self._default_encoder self.roi = self._default_roi self.laser_detector = self._default_laser_detector self.video_processor = VideoProcessor(self.camera, self.encoder, self.roi, self.laser_detector)
def set_region_of_interest_from_rel_points(self, x_rel, y_rel, w_rel, h_rel): self.roi = ROI(x_rel, y_rel, w_rel, h_rel) self.video_processor.roi = self.roi
def setUp(self): self.img2point = Mock() self.img2point.get_points.return_value = np.array([1.0, 1.0, 1.0]) self.roi = ROI(0, 0, 1, 1) self.laser_theta = .77
def test_get_returns_roi_of_expected_size(self): test_array = self.tarray() roi = ROI(0, 0, 1, 1) self.assertTrue((roi.get(test_array) == test_array).all())
def test_get_points_should_return_a_list_of_points(self): roi = ROI(0, 0, 1, 1) self.assertEquals([0, 0, 1, 1], roi.get_points())