Merge pull request #1370 from r3vl1s/imread_collections

load correct _frame_index
This commit is contained in:
Steven Silvester
2015-02-05 20:03:17 -06:00
2 changed files with 8 additions and 1 deletions
+1 -1
View File
@@ -254,7 +254,7 @@ class ImageCollection(object):
if ((self.conserve_memory and n != self._cached) or
(self.data[idx] is None)):
if self._frame_index:
fname, img_num = self._frame_index[idx]
fname, img_num = self._frame_index[n]
self.data[idx] = self.load_func(fname, img_num=img_num,
**self.load_func_kwargs)
else:
+7
View File
@@ -50,6 +50,10 @@ class TestMultiImage():
assert type(img[i]) is np.ndarray
assert_allclose(img[0], img[-num])
assert_raises(AssertionError,
assert_allclose,
img[0], img[1])
# assert_raises expects a callable, hence this thin wrapper function.
def return_img(n):
return img[n]
@@ -77,6 +81,9 @@ class TestMultiImage():
def test_concatenate(self):
for img in self.imgs:
if img[0].shape != img[-1].shape:
assert_raises(ValueError, img.concatenate)
continue
array = img.concatenate()
assert_equal(array.shape, (len(img),) + img[0].shape)