def test_subsample(self):
narys = 3
arys, sh, sz = _generateTestArrays(narys)
sampFactors = [2, (2, 3, 3)]
def subsamp(ary, factor):
if not hasattr(factor, "__len__"):
factor = [factor] * ary.ndim
slices = [
slice(0, ary.shape[i], factor[i]) for i in xrange(ary.ndim)
]
return ary[slices]
imageData = ImagesLoader(self.sc).fromArrays(arys)
for sampFactor in sampFactors:
subsampData = imageData.subsample(sampFactor)
expectedArys = map(lambda ary: subsamp(ary, sampFactor), arys)
subsampled = subsampData.collect()
for actual, expected in zip(subsampled, expectedArys):
assert_true(array_equal(expected, actual[1]))
assert_equals(tuple(expectedArys[0].shape), subsampled[0][1].shape)
assert_equals(tuple(expectedArys[0].shape),
subsampData._dims.count)
assert_equals(str(arys[0].dtype), str(subsampled[0][1].dtype))
assert_equals(str(subsampled[0][1].dtype), subsampData._dtype)
def test_subsample(self):
narys = 3
arys, sh, sz = _generateTestArrays(narys)
sampFactors = [2, (2, 3, 3)]
def subsamp(ary, factor):
if not hasattr(factor, "__len__"):
factor = [factor] * ary.ndim
slices = [slice(0, ary.shape[i], factor[i]) for i in xrange(ary.ndim)]
return ary[slices]
imageData = ImagesLoader(self.sc).fromArrays(arys)
for sampFactor in sampFactors:
subsampData = imageData.subsample(sampFactor)
expectedArys = map(lambda ary: subsamp(ary, sampFactor), arys)
subsampled = subsampData.collect()
for actual, expected in zip(subsampled, expectedArys):
assert_true(array_equal(expected, actual[1]))
assert_equals(tuple(expectedArys[0].shape), subsampled[0][1].shape)
assert_equals(tuple(expectedArys[0].shape), subsampData._dims.count)
assert_equals(str(arys[0].dtype), str(subsampled[0][1].dtype))
assert_equals(str(subsampled[0][1].dtype), subsampData._dtype)
