def test_PickleLoadObject(): ''' Test for PickleLoadObject object ''' tr = TrackResults(500, 1, 'l1ca') for i in range(500): tr.ms_tracked[i] = i tr.absolute_sample[i] = i tr.dump('test_output.bin', 500) for i in range(500): tr.ms_tracked[i] = i + 500 tr.absolute_sample[i] = i + 500 tr.dump('test_output.bin', 500) loadObj = PickleLoadObject('test_output.PRN-2.l1ca.bin.track_results') it = iter(loadObj) o0 = it.next() o1 = it.next() try: it.next() assert False except StopIteration: pass try: it.next() assert False except StopIteration: pass assert isinstance(o0, TrackResults) for i in range(500): assert o0.ms_tracked[i] == i assert o0.absolute_sample[i] == i assert isinstance(o1, TrackResults) for i in range(500): assert o1.ms_tracked[i] == i + 500 assert o1.absolute_sample[i] == i + 500
def __testSetup(): ''' Test utility ''' removeTrackingOutputFiles("test_output.bin") tr1 = TrackResults(500, 0, 'l1ca') for i in range(500): tr1.ms_tracked[i] = i * 2 tr1.absolute_sample[i] = i * 2 tr1.status = 'A' tr1.dump('test_output.bin', 500) tr2 = TrackResults(500, 1, 'l1ca') for i in range(500): tr2.ms_tracked[i] = i * 2 + 1 tr2.absolute_sample[i] = i * 2 + 1 tr2.status = 'B' tr2.dump('test_output.bin', 500)
def test_TrackResultsFile(): ''' Test for TrackResults object ''' tr = TrackResults(500, 1, 'l1ca') for i in range(500): tr.ms_tracked[i] = i tr.absolute_sample[i] = i tr.status = 'A' tr.dump('test_output.bin', 500) for i in range(500): tr.ms_tracked[i] = i + 500 tr.absolute_sample[i] = i + 500 tr.status = 'B' tr.dump('test_output.bin', 500) obj = TrackResultFile( PickleLoadObject('test_output.PRN-2.l1ca.bin.track_results')) it = iter(obj) for i in range(500): o, idx = it.next() assert o.status == 'A' assert idx == i assert o.ms_tracked[i] == i assert o.absolute_sample[i] == i for i in range(500): o, idx = it.next() assert o.status == 'B' assert idx == i assert o.ms_tracked[i] == i + 500 assert o.absolute_sample[i] == i + 500 try: it.next() assert False except StopIteration: pass try: it.next() assert False except StopIteration: pass