def test_detect_returns_an_empty_matrix_with_nothing_in_range(self): test = np.ones((100, 100, 3), dtype='uint8') * 255 expected = np.zeros((100, 100, 1), dtype='uint8') laser_detector = LaserDetector((128, 128, 128), (129, 129, 129)) result = laser_detector.detect(test) self.assertTrue((expected == result).all())
def test_detect_returns_matrix_of_matched_points_if_in_range(self): test = np.ones((10, 10, 3), dtype='uint8') * 129 expected = np.ones((10, 10, 1), dtype='uint8') * 255 laser_detector = LaserDetector((128, 128, 128), (130, 130, 130)) result = laser_detector.detect(test) self.assertTrue((expected == result).all(), "{} != {}".format(expected, result)) self.assertEquals(expected.dtype, result.dtype)