def test_meanByFixedLength(self): rdd = self.sc.parallelize([((0,), array([0, 1, 2, 3, 4, 5, 6, 7], dtype='float16'))]) data = Series(rdd) test1 = data.meanByFixedLength(4) assert(test1.keys().collect() == [(0,)]) assert(allclose(test1.index, array([0, 1, 2, 3]))) assert(allclose(test1.values().collect(), [[2, 3, 4, 5]])) test2 = data.meanByFixedLength(2) assert(test2.keys().collect() == [(0,)]) assert(allclose(test2.index, array([0, 1]))) assert(allclose(test2.values().collect(), [[3, 4]]))