def test_overlay_encoder_places_indicator_correct_place(self):
encoder = Encoder(point=[0.5, 0.5])
encoder.process(self.blackimage)
self.small_black_image = np.zeros((50, 50, 3))
resulting_image = encoder.overlay_encoder(self.small_black_image)
# Encode indicator is a circle centered on point with some decoration, just checking for one point
self.assertTrue((resulting_image[27][27] == [0, 0, 255]).all())
def test_process_given_alternating_BW_counts_only_on_change(self):
expected = [1, 0, 1, 0]
encoder = Encoder()
actual = [encoder.process(image) for image in [self.whiteimage, self.whiteimage, self.blackimage, self.blackimage]]
self.assertEqual(expected, actual)
def test_overlay_encoder_places_encoder_indicator_correct_place_if_wont_fit(
self):
encoder = Encoder(point=[0.02, 0.02])
encoder.process(self.blackimage)
resulting_image = encoder.overlay_encoder(self.blackimage)
# Encode indicator is a circle centered on point with some decoration, just checking for one point
self.assertTrue((resulting_image[4][4] == [0, 0, 255]).all())
def test_should_capture_frame_for_section_returns_true_and_current_index_given_change_detected(self):
encoder = Encoder(threshold=450, null_zone=50)
should_capture, idx = encoder.should_capture_frame_for_section(self.whiteimage)
self.assertTrue(should_capture)
should_capture, idx = encoder.should_capture_frame_for_section(self.whiteimage)
self.assertFalse(should_capture)
def test_overlays_threshold_history_shows_threshold_and_null_lines(self):
blackimage = np.ones((255, 255, 3), dtype='uint8') * 10
greyimage = np.ones((255, 255, 3), dtype='uint8') * 100
whiteimage = np.ones((255, 255, 3), dtype='uint8') * 200
encoder = Encoder(point=[0.5, 0.5], threshold=300, null_zone=150, history_length=10)
for image in [
blackimage, blackimage, blackimage, blackimage,
greyimage, greyimage, greyimage, greyimage,
whiteimage, whiteimage, whiteimage, whiteimage
]:
encoder.process(image)
resulting_image = encoder.overlay_history(image)
self.assertTrue((resulting_image[255-10][0] == [0, 0, 255]).all())
self.assertFalse((resulting_image[255-11][0] == [0, 0, 255]).all())
self.assertTrue((resulting_image[255-10][1] == [0, 0, 255]).all())
self.assertFalse((resulting_image[255-11][1] == [0, 0, 255]).all())
self.assertTrue((resulting_image[255-100][2] == [0, 255, 255]).all())
self.assertTrue((resulting_image[255-100][3] == [0, 255, 255]).all())
self.assertTrue((resulting_image[255-100][4] == [0, 255, 255]).all())
self.assertTrue((resulting_image[255-100][5] == [0, 255, 255]).all())
self.assertFalse((resulting_image[255-101][5] == [0, 255, 255]).all())
self.assertTrue((resulting_image[255-200][6] == [0, 255, 0]).all())
self.assertTrue((resulting_image[255-200][7] == [0, 255, 0]).all())
self.assertTrue((resulting_image[255-200][8] == [0, 255, 0]).all())
self.assertTrue((resulting_image[255-200][9] == [0, 255, 0]).all())
self.assertFalse((resulting_image[255-201][9] == [0, 255, 0]).all())
def test_process_given_alternating_BW_adds_expected_degrees(self):
expected = [1, 1, 1, 1]
encoder = Encoder()
actual = [encoder.process(image) for image in [self.whiteimage, self.blackimage, self.whiteimage, self.blackimage, ]]
self.assertEqual(expected, actual)
def test_overlay_history_shows_threshold_and_null_lines(self):
image = np.ones((255, 255, 3), dtype='uint8') * 10
encoder = Encoder(point=[0.5, 0.5], threshold=300, null_zone=150)
encoder.process(image)
resulting_image = encoder.overlay_history(image)
self.assertTrue((resulting_image[255-150][9] == [255, 255, 255]).all())
self.assertTrue((resulting_image[255-50][9] == [255, 255, 255]).all())
def test_overlays_threshold_history_shows_threshold_and_null_lines(self):
blackimage = np.ones((255, 255, 3), dtype='uint8') * 10
greyimage = np.ones((255, 255, 3), dtype='uint8') * 100
whiteimage = np.ones((255, 255, 3), dtype='uint8') * 200
encoder = Encoder(point=[0.5, 0.5],
threshold=300,
null_zone=150,
history_length=10)
for image in [
blackimage, blackimage, blackimage, blackimage, greyimage,
greyimage, greyimage, greyimage, whiteimage, whiteimage,
whiteimage, whiteimage
]:
encoder.process(image)
resulting_image = encoder.overlay_history(image)
self.assertTrue((resulting_image[255 - 10][0] == [0, 0, 255]).all())
self.assertFalse((resulting_image[255 - 11][0] == [0, 0, 255]).all())
self.assertTrue((resulting_image[255 - 10][1] == [0, 0, 255]).all())
self.assertFalse((resulting_image[255 - 11][1] == [0, 0, 255]).all())
self.assertTrue((resulting_image[255 - 100][2] == [0, 255, 255]).all())
self.assertTrue((resulting_image[255 - 100][3] == [0, 255, 255]).all())
self.assertTrue((resulting_image[255 - 100][4] == [0, 255, 255]).all())
self.assertTrue((resulting_image[255 - 100][5] == [0, 255, 255]).all())
self.assertFalse((resulting_image[255 - 101][5] == [0, 255,
255]).all())
self.assertTrue((resulting_image[255 - 200][6] == [0, 255, 0]).all())
self.assertTrue((resulting_image[255 - 200][7] == [0, 255, 0]).all())
self.assertTrue((resulting_image[255 - 200][8] == [0, 255, 0]).all())
self.assertTrue((resulting_image[255 - 200][9] == [0, 255, 0]).all())
self.assertFalse((resulting_image[255 - 201][9] == [0, 255, 0]).all())
def test_overlay_encoder_places_indicator_correct_place(self):
encoder = Encoder(point=[0.5, 0.5])
encoder.process(self.blackimage)
self.small_black_image = np.zeros((50, 50, 3))
resulting_image = encoder.overlay_encoder(self.small_black_image)
# Encode indicator is a circle centered on point with some decoration, just checking for one point
self.assertTrue((resulting_image[27][27] == [0, 0, 255]).all())
def test_overlay_encoder_places_indicator_correct_place_on_black_mask(self):
encoder = Encoder(point=[0.5, 0.5])
encoder.process(self.whiteimage)
self.small_white_image = np.ones((50, 50, 3)) * 255
resulting_image = encoder.overlay_encoder(self.small_white_image)
# Encode indicator is a circle centered on point with some decoration, just checking for one point
self.assertTrue((resulting_image[0][0] == [0, 0, 0]).all())
def test_overlay_encoder_places_indicator_correct_place_on_black_mask(
self):
encoder = Encoder(point=[0.5, 0.5])
encoder.process(self.whiteimage)
self.small_white_image = np.ones((50, 50, 3)) * 255
resulting_image = encoder.overlay_encoder(self.small_white_image)
# Encode indicator is a circle centered on point with some decoration, just checking for one point
self.assertTrue((resulting_image[0][0] == [0, 0, 0]).all())
def test_should_capture_frame_for_section_should_reset_to_zero_if_number_of_sections_exceeds_total_sections(self):
encoder = Encoder(threshold=450, null_zone=50, sections=2)
encoder.should_capture_frame_for_section(self.blackimage)
encoder.should_capture_frame_for_section(self.whiteimage)
retVal, rotation = encoder.should_capture_frame_for_section(self.blackimage)
self.assertTrue(retVal)
self.assertEqual(0, rotation)
def test_process_given_alternating_BW_adds_expected_degrees_within_threshold(self):
blackimage = np.ones((100, 100, 3), dtype='uint8') * 100
expected = [1, 0, 1, 0, 0, 1, 0, 1]
encoder = Encoder(threshold=450, null_zone=50)
actual = [encoder.process(image) for image in [self.whiteimage, self.nullimage, blackimage, self.nullimage, self.nullimage, self.whiteimage, self.nullimage, blackimage, ]]
self.assertEqual(expected, actual)
def test_process_given_alternating_BW_counts_at_specific_point(self):
whiteimage = np.zeros((100, 100, 3), dtype='uint8')
whiteimage[4][4] = [255, 255, 255]
expected = [1, 1, 1, 1]
encoder = Encoder(point=[0.04, 0.04])
actual = [encoder.process(image) for image in [whiteimage, self.blackimage, whiteimage, self.blackimage, ]]
self.assertEqual(expected, actual)
def test_overlay_history_shows_threshold_and_null_lines(self):
image = np.ones((255, 255, 3), dtype='uint8') * 10
encoder = Encoder(point=[0.5, 0.5], threshold=300, null_zone=150)
encoder.process(image)
resulting_image = encoder.overlay_history(image)
self.assertTrue((resulting_image[255 - 150][9] == [255, 255,
255]).all())
self.assertTrue((resulting_image[255 - 50][9] == [255, 255,
255]).all())
def test_should_capture_frame_for_section_should_reset_to_zero_if_number_of_sections_exceeds_total_sections(
self):
encoder = Encoder(threshold=450, null_zone=50, sections=2)
encoder.should_capture_frame_for_section(self.blackimage)
encoder.should_capture_frame_for_section(self.whiteimage)
retVal, rotation = encoder.should_capture_frame_for_section(
self.blackimage)
self.assertTrue(retVal)
self.assertEqual(0, rotation)
def test_should_capture_frame_for_section_returns_true_and_current_index_given_change_detected(
self):
encoder = Encoder(threshold=450, null_zone=50)
should_capture, idx = encoder.should_capture_frame_for_section(
self.whiteimage)
self.assertTrue(should_capture)
should_capture, idx = encoder.should_capture_frame_for_section(
self.whiteimage)
self.assertFalse(should_capture)
def test_thrshold_changes_threshold(self):
blackimage = np.ones((100, 100, 3), dtype='uint8') * 100
expected = [1, 0, 1, 0, 0, 1, 0, 1]
encoder = Encoder(threshold=382, null_zone=382)
encoder.threshold = 450
encoder.null_zone = 50
actual = [encoder.process(image) for image in [self.whiteimage, self.nullimage, blackimage, self.nullimage, self.nullimage, self.whiteimage, self.nullimage, blackimage, ]]
self.assertEqual(expected, actual)
def test_process_given_alternating_BW_counts_only_on_change(self):
expected = [1, 0, 1, 0]
encoder = Encoder()
actual = [
encoder.process(image) for image in [
self.whiteimage, self.whiteimage, self.blackimage,
self.blackimage
]
]
self.assertEqual(expected, actual)
def test_overlay_encoder_indicator_color_is_correct(self):
greyimage = np.ones((100, 100, 3), dtype='uint8') * 128
encoder = Encoder(point=[0.5, 0.5], threshold=384, null_zone=50)
encoder.process(self.blackimage)
resulting_image = encoder.overlay_encoder(self.blackimage)
self.assertTrue((resulting_image[52][52] == [0, 0, 255]).all())
encoder.process(greyimage)
resulting_image = encoder.overlay_encoder(self.blackimage)
self.assertTrue((resulting_image[52][52] == [0, 255, 255]).all())
encoder.process(self.whiteimage)
resulting_image = encoder.overlay_encoder(self.blackimage)
self.assertTrue((resulting_image[52][52] == [0, 255, 0]).all())
def test_process_given_alternating_BW_adds_expected_degrees(self):
expected = [1, 1, 1, 1]
encoder = Encoder()
actual = [
encoder.process(image) for image in [
self.whiteimage,
self.blackimage,
self.whiteimage,
self.blackimage,
]
]
self.assertEqual(expected, actual)
def test_process_given_alternating_BW_counts_at_specific_point(self):
whiteimage = np.zeros((100, 100, 3), dtype='uint8')
whiteimage[4][4] = [255, 255, 255]
expected = [1, 1, 1, 1]
encoder = Encoder(point=[0.04, 0.04])
actual = [
encoder.process(image) for image in [
whiteimage,
self.blackimage,
whiteimage,
self.blackimage,
]
]
self.assertEqual(expected, actual)
def test_process_given_alternating_BW_adds_expected_degrees_within_threshold(
self):
blackimage = np.ones((100, 100, 3), dtype='uint8') * 100
expected = [1, 0, 1, 0, 0, 1, 0, 1]
encoder = Encoder(threshold=450, null_zone=50)
actual = [
encoder.process(image) for image in [
self.whiteimage,
self.nullimage,
blackimage,
self.nullimage,
self.nullimage,
self.whiteimage,
self.nullimage,
blackimage,
]
]
self.assertEqual(expected, actual)
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 test_thrshold_changes_threshold(self):
blackimage = np.ones((100, 100, 3), dtype='uint8') * 100
expected = [1, 0, 1, 0, 0, 1, 0, 1]
encoder = Encoder(threshold=382, null_zone=382)
encoder.threshold = 450
encoder.null_zone = 50
actual = [
encoder.process(image) for image in [
self.whiteimage,
self.nullimage,
blackimage,
self.nullimage,
self.nullimage,
self.whiteimage,
self.nullimage,
blackimage,
]
]
self.assertEqual(expected, actual)
def test_overlay_encoder_places_encoder_indicator_correct_place_if_wont_fit(self):
encoder = Encoder(point=[0.02, 0.02])
encoder.process(self.blackimage)
resulting_image = encoder.overlay_encoder(self.blackimage)
# Encode indicator is a circle centered on point with some decoration, just checking for one point
self.assertTrue((resulting_image[4][4] == [0, 0, 255]).all())
def test_overlay_encoder_indicator_color_is_correct(self):
greyimage = np.ones((100, 100, 3), dtype='uint8') * 128
encoder = Encoder(point=[0.5, 0.5], threshold=384, null_zone=50)
encoder.process(self.blackimage)
resulting_image = encoder.overlay_encoder(self.blackimage)
self.assertTrue((resulting_image[52][52] == [0, 0, 255]).all())
encoder.process(greyimage)
resulting_image = encoder.overlay_encoder(self.blackimage)
self.assertTrue((resulting_image[52][52] == [0, 255, 255]).all())
encoder.process(self.whiteimage)
resulting_image = encoder.overlay_encoder(self.blackimage)
self.assertTrue((resulting_image[52][52] == [0, 255, 0]).all())
def configure_encoder(self, point, threshold, null_zone, sections):
self.encoder = Encoder(point, threshold, null_zone, 20, sections)
self.video_processor.encoder = self.encoder
