def test_flatten_considers_files(self):
unflattened_chunks = [
{'filenum': 0, 'begin': 200, 'end': 300, 't': 10.0, 'peeraddr': 'A'},
{'filenum': 1, 'begin': 300, 'end': 400, 't': 10.0, 'peeraddr': 'A'}
]
expected_result = unflattened_chunks
log = TrLog()
log.chunks = unflattened_chunks
log.flatten()
self.assertEquals(expected_result, log.chunks)
def test_flatten_handles_interwoven_peers(self):
unflattened_chunks = [
{'filenum': 0, 'begin': 200, 'end': 300, 't': 10.0, 'peeraddr': 'A'},
{'filenum': 0, 'begin': 500, 'end': 600, 't': 10.0, 'peeraddr': 'B'},
{'filenum': 0, 'begin': 300, 'end': 400, 't': 10.0, 'peeraddr': 'A'},
{'filenum': 0, 'begin': 600, 'end': 700, 't': 10.0, 'peeraddr': 'B'},
]
expected_result = [
{'filenum': 0, 'begin': 200, 'end': 400, 't': 10.0, 'peeraddr': 'A'},
{'filenum': 0, 'begin': 500, 'end': 700, 't': 10.0, 'peeraddr': 'B'},
]
log = TrLog()
log.chunks = unflattened_chunks
log.flatten()
self.assertEquals(expected_result, log.chunks)
