Skip to content

O
opencv
Commit 56f17e49 authored by Roman Donchenko's avatar Roman Donchenko
Browse files
Options

test2.py: fail if a downloaded image can't be decoded

parent 293ea03d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
modules/python/test/test2.py
View file @ 56f17e49
...@@ -13,7 +13,9 @@ class NewOpenCVTests(unittest.TestCase): ...@@ -13,7 +13,9 @@ class NewOpenCVTests(unittest.TestCase):
def get_sample(self, filename, iscolor = cv.CV_LOAD_IMAGE_COLOR): def get_sample(self, filename, iscolor = cv.CV_LOAD_IMAGE_COLOR):
if not filename in self.image_cache: if not filename in self.image_cache:
filedata = urllib.urlopen("https://raw.github.com/Itseez/opencv/2.4/" + filename).read() filedata = urllib.urlopen("https://raw.github.com/Itseez/opencv/2.4/" + filename).read()
self.image_cache[filename] = cv2.imdecode(np.fromstring(filedata, dtype=np.uint8), iscolor) image = cv2.imdecode(np.fromstring(filedata, dtype=np.uint8), iscolor)
self.assertFalse(image is None)
self.image_cache[filename] = image
return self.image_cache[filename] return self.image_cache[filename]
def setUp(self): def setUp(self):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment