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