generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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

Browse Source
This commit is contained in:
Roman Donchenko 2015-09-17 18:17:06 +03:00
parent 293ea03dcc
commit 56f17e4921
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/python/test/test2.py
View File

@ -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):