def test_stats(self): from test_utils import elementwiseMean, elementwiseVar arys, shape, size = _generateTestArrays(2, 'uint8') imageData = ImagesLoader(self.sc).fromArrays(arys) statsval = imageData.stats() floatarys = [ary.astype('float16') for ary in arys] # StatsCounter contains a few different measures, only test a couple: expectedMean = elementwiseMean(floatarys) expectedVar = elementwiseVar(floatarys) assert_true(allclose(expectedMean, statsval.mean())) assert_true(allclose(expectedVar, statsval.variance()))
def test_stats(self): from test_utils import elementwiseMean, elementwiseVar arys, shape, size = _generateTestArrays(2, 'uint8') imageData = ImagesLoader(self.sc).fromArrays(arys) statsval = imageData.stats() floatarys = [ary.astype('float16') for ary in arys] # StatsCounter contains a few different measures, only test a couple: expectedMean = elementwiseMean(floatarys) expectedVar = elementwiseVar(floatarys) assert_true(allclose(expectedMean, statsval.mean())) assert_true(allclose(expectedVar, statsval.variance()))