def test_mean(self): from test_utils import elementwiseMean series = self.generateTestSeries() meanVal = series.mean() expected = elementwiseMean(series.values().collect()) assert_true(allclose(expected, meanVal)) assert_equals('float64', str(meanVal.dtype))
def test_mean(self): from test_utils import elementwiseMean arys, shape, size = _generateTestArrays(2, 'uint8') imageData = ImagesLoader(self.sc).fromArrays(arys) meanVal = imageData.mean() expected = elementwiseMean(arys).astype('float16') assert_true(allclose(expected, meanVal)) assert_equals('float64', str(meanVal.dtype))
def test_stats(self): from test_utils import elementwiseMean, elementwiseVar series = self.generateTestSeries() statsVal = series.stats() arys = series.values().collect() floatArys = [ary.astype('float16') for ary in arys] 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()))