def test_get_blocks_single_block(self): stream = self._lzo_stream(1) index = lzo_indexer.get_lzo_blocks(stream) block = index.next() self.assertEqual(block, 38) with self.assertRaises(StopIteration): index.next()
def test_get_blocks_multiple_blocks(self): stream = self._lzo_stream(10**6) index = lzo_indexer.get_lzo_blocks(stream) expected_offsets = [38, 1233, 2428, 3623] self.assertEqual(list(index), expected_offsets)