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