def test_sortingpipe(self): iterations = 100 forward = sortingPipe() reverse = sortingPipe(reverse=True) [forward.push(random.random()) for _ in xrange(iterations)] [reverse.push(random.random()) for _ in xrange(iterations)] self.assertTrue(is_sorted(forward)) self.assertTrue(is_sorted(reverse, reverse=True))
def __init__(self, file_obj, reverse=False): ''' IndexedWriter takes ownership of file_obj (closes upon close()) ''' self.file_obj = file_obj self.pickler = cPickle.Pickler(self.file_obj, cPickle.HIGHEST_PROTOCOL) self._value_file_obj = tempfile.TemporaryFile() self._value_pickler = cPickle.Pickler(self._value_file_obj, cPickle.HIGHEST_PROTOCOL) self._index = sortingPipe(key=operator.itemgetter(0), reverse=reverse)