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