def test_squelch(self): rdd = self.sc.parallelize([(0, array([1, 2])), (0, array([3, 4]))]) data = Series(rdd) squelched = data.squelch(5) assert(allclose(squelched.collectValuesAsArray(), [[0, 0], [0, 0]])) squelched = data.squelch(3) assert(allclose(squelched.collectValuesAsArray(), [[0, 0], [3, 4]])) squelched = data.squelch(1) assert(allclose(squelched.collectValuesAsArray(), [[1, 2], [3, 4]]))