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_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)
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)
