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())
