Where we deal with .pyc files, also deal with .pyo files. Fixes #195.

Author: nedbat

CHANGES.txt

@@ -18,6 +18,9 @@ Version 3.5.3b1
 
 - Fixed memory leaks under Python 3, thanks, Brett Cannon.
 
+- Optimized .pyo files may not have been handled correctly, `issue 195`_.
+  Thanks, Marius Gedminas.
+
 - Try to do a better job of the impossible task of detecting when we can't
   build the C extension, fixing `issue 183`_.
 
@@ -26,9 +29,11 @@ Version 3.5.3b1
 .. _issue 82: https://bitbucket.org/ned/coveragepy/issue/82/tokenerror-when-generating-html-report
 .. _issue 179: https://bitbucket.org/ned/coveragepy/issue/179/htmlreporter-fails-when-source-file-is
 .. _issue 183: https://bitbucket.org/ned/coveragepy/issue/183/install-fails-for-python-23
+.. _issue 195: https://bitbucket.org/ned/coveragepy/issue/195/pyo-file-handling-in-codeunit
 .. _tox: http://tox.readthedocs.org/
 
 
+
 Version 3.5.2 --- 4 May 2012
 ----------------------------
 

coverage/codeunit.py

@@ -52,7 +52,7 @@ def __init__(self, morf, file_locator):
         else:
             f = morf
         # .pyc files should always refer to a .py instead.
-        if f.endswith('.pyc'):
+        if f.endswith('.pyc') or f.endswith('.pyo'):
             f = f[:-1]
         self.filename = self.file_locator.canonical_filename(f)