def test_between(self): rdd = self.sc.parallelize([(0, array([4, 5, 6, 7])), (1, array([8, 9, 10, 11]))]) data = Series(rdd).between(0, 1) assert(allclose(data.index, array([0, 1]))) assert(allclose(data.first()[1], array([4, 5])))