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