def test_opencv_numpy_images_are_on_the_whitelist_of_types(self): color_image = cv2.imread(path.join(_this_path, 'data', 'object.png')) gray_image = cv2.cvtColor(color_image, cv2.COLOR_BGR2GRAY) d = ImageDict() self.assertTrue(d._is_on_whitelist(color_image)) self.assertTrue(d._is_on_whitelist(gray_image))