diff options
| -rw-r--r-- | Makefile | 1 | ||||
| -rw-r--r-- | coverage/bytecode.py | 8 | ||||
| -rw-r--r-- | test/test_coverage.py | 1 |
3 files changed, 5 insertions, 5 deletions
@@ -31,6 +31,7 @@ LINTABLE = \ test/test_api.py \ test/test_cmdline.py \ test/test_codeunit.py \ + test/test_coverage.py \ test/test_data.py \ test/test_execfile.py \ test/test_farm.py \ diff --git a/coverage/bytecode.py b/coverage/bytecode.py index 0680a57..82d327c 100644 --- a/coverage/bytecode.py +++ b/coverage/bytecode.py @@ -23,10 +23,10 @@ class ByteCodes(object): self.offset = 0 if sys.hexversion > 0x03000000: - def byte_at(self, i): + def __getitem__(self, i): return self.code[i] else: - def byte_at(self, i): + def __getitem__(self, i): return ord(self.code[i]) def __iter__(self): @@ -37,12 +37,12 @@ class ByteCodes(object): raise StopIteration bc = ByteCode() - bc.op = self.byte_at(self.offset) + bc.op = self[self.offset] bc.offset = self.offset next_offset = self.offset+1 if bc.op >= opcode.HAVE_ARGUMENT: - bc.oparg = self.byte_at(self.offset+1) + 256*self.byte_at(self.offset+2) + bc.oparg = self[self.offset+1] + 256*self[self.offset+2] next_offset += 2 label = -1 diff --git a/test/test_coverage.py b/test/test_coverage.py index 4bbace5..14ef27e 100644 --- a/test/test_coverage.py +++ b/test/test_coverage.py @@ -5,7 +5,6 @@ import os, sys, unittest import coverage -from coverage.backward import StringIO coverage.use_cache(0) sys.path.insert(0, os.path.split(__file__)[0]) # Force relative import for Py3k |