def testBulkreader(tmpdir): q = PriorityEnqueue(str(tmpdir)) q.queue([dict(id=i, p='L', expected=3) for i in xrange(2000)]) q.close() # count how many files it created fns = tmpdir.join('3').listdir() nfiles = len(fns) dq = PriorityDequeue(str(tmpdir)) count = 0 while nfiles > 0: reader = dq.bulkreader() assert reader is not None for curi in reader: count += 1 nfiles -= 1 assert count == 2000 reader = dq.bulkreader() assert reader is None