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