Ejemplo n.º 1
0
    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))
Ejemplo n.º 2
0
 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)