def test_a_colour_can_be_changed(self): img = cv2.imread(os.path.join(self.test_data_path,'GreenThresholdTest.png'),1) threshold = 20 initial_expected = [0,0,0,-1,-1] initial_colour = [128,128,128] new_expected = [-1,-1,-1,-1,-1] new_colour = [64,64,64] mapper = Mapper(initial_colour, threshold) initial_result = mapper.get_points(img) self.assertEquals(initial_expected,initial_result) mapper.set_colour(new_colour) new_result = mapper.get_points(img) self.assertEquals(new_expected,new_result)
def test_a_colour_can_be_changed(self): img = cv2.imread( os.path.join(self.test_data_path, 'GreenThresholdTest.png'), 1) threshold = 20 initial_expected = [0, 0, 0, -1, -1] initial_colour = [128, 128, 128] new_expected = [-1, -1, -1, -1, -1] new_colour = [64, 64, 64] mapper = Mapper(initial_colour, threshold) initial_result = mapper.get_points(img) self.assertEquals(initial_expected, initial_result) mapper.set_colour(new_colour) new_result = mapper.get_points(img) self.assertEquals(new_expected, new_result)
def test_given_a_threshold_items_in_threshold_work_for_blue(self): img = cv2.imread(os.path.join(self.test_data_path,'BlueThresholdTest.png'),1) threshold = 20 expected = [0,0,0,-1,-1] colour = [128,128,128] mapper = Mapper(colour, threshold) actual = mapper.get_points(img) self.assertEquals(expected,actual)
def test_given_an_colour_image_and_specific_colour_a_point_map_returned(self): img = cv2.imread(os.path.join(self.test_data_path,'SimpleTestImage2.png'),1) expected = [i for i in range(0,20)] colour = [255,128,0] threshold = 0 mapper = Mapper(colour,threshold) actual = mapper.get_points(img) self.assertEquals(expected,actual)
def test_given_an_image_with_no_points_a_point_map_returned(self): img = cv2.imread(os.path.join(self.test_data_path,'SimpleTestImage5.png'),1) expected = [-1 for i in range(0,20)] colour = [255,255,255] threshold = 0 mapper = Mapper(colour,threshold) actual = mapper.get_points(img) self.assertEquals(expected,actual)
def test_given_a_threshold_items_in_threshold_work_for_blue(self): img = cv2.imread( os.path.join(self.test_data_path, 'BlueThresholdTest.png'), 1) threshold = 20 expected = [0, 0, 0, -1, -1] colour = [128, 128, 128] mapper = Mapper(colour, threshold) actual = mapper.get_points(img) self.assertEquals(expected, actual)
def test_given_an_image_with_no_points_a_point_map_returned(self): img = cv2.imread( os.path.join(self.test_data_path, 'SimpleTestImage5.png'), 1) expected = [-1 for i in range(0, 20)] colour = [255, 255, 255] threshold = 0 mapper = Mapper(colour, threshold) actual = mapper.get_points(img) self.assertEquals(expected, actual)
def test_a_threshold_can_be_changed(self): img = cv2.imread(os.path.join(self.test_data_path,'GreenThresholdTest.png'),1) initial_threshold = 20 new_threshold = 21 expected = [0,0,0,0,0] colour = [128,128,128] mapper = Mapper(colour, initial_threshold) mapper.set_threshold(new_threshold) actual = mapper.get_points(img) self.assertEquals(expected,actual)
def test_given_an_colour_image_and_specific_colour_a_point_map_returned( self): img = cv2.imread( os.path.join(self.test_data_path, 'SimpleTestImage2.png'), 1) expected = [i for i in range(0, 20)] colour = [255, 128, 0] threshold = 0 mapper = Mapper(colour, threshold) actual = mapper.get_points(img) self.assertEquals(expected, actual)
def test_a_threshold_can_be_changed(self): img = cv2.imread( os.path.join(self.test_data_path, 'GreenThresholdTest.png'), 1) initial_threshold = 20 new_threshold = 21 expected = [0, 0, 0, 0, 0] colour = [128, 128, 128] mapper = Mapper(colour, initial_threshold) mapper.set_threshold(new_threshold) actual = mapper.get_points(img) self.assertEquals(expected, actual)