From 5e5cf2d5b9d7decfce16142a7cf7cc140fcbf354 Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Sat, 28 Sep 2013 10:07:55 -0400 Subject: More abstractions for bytes objects. Cleans up some version checks in the real code. --- coverage/bytecode.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) (limited to 'coverage/bytecode.py') diff --git a/coverage/bytecode.py b/coverage/bytecode.py index 06bc1dfd..85360638 100644 --- a/coverage/bytecode.py +++ b/coverage/bytecode.py @@ -1,6 +1,8 @@ """Bytecode manipulation for coverage.py""" -import opcode, sys, types +import opcode, types + +from coverage.backward import byte_to_int class ByteCode(object): """A single bytecode.""" @@ -31,12 +33,8 @@ class ByteCodes(object): def __init__(self, code): self.code = code - if sys.version_info >= (3, 0): - def __getitem__(self, i): - return self.code[i] - else: - def __getitem__(self, i): - return ord(self.code[i]) + def __getitem__(self, i): + return byte_to_int(self.code[i]) def __iter__(self): offset = 0 -- cgit v1.2.1