diff options
author | Ned Batchelder <nedbat@gmail.com> | 2017-01-14 17:18:56 -0500 |
---|---|---|
committer | Ned Batchelder <nedbat@gmail.com> | 2017-01-14 17:18:56 -0500 |
commit | 3d8e48162df541de153ee0f23f75074e2fc13376 (patch) | |
tree | fd22302b96422e6749f8beb5d376dff03b3a66cf /tests/test_python.py | |
parent | 6d24c0864b2e18fb646ddca1cefd71819b64e83d (diff) | |
parent | d9ca8593eba371b49402974c1c1f2bf7fd52fed9 (diff) | |
download | python-coveragepy-3d8e48162df541de153ee0f23f75074e2fc13376.tar.gz |
Merged in dachary/coverage.py/issue-426 (pull request #122)
make --source module do the same as --source directory #426
Diffstat (limited to 'tests/test_python.py')
-rw-r--r-- | tests/test_python.py | 27 |
1 files changed, 26 insertions, 1 deletions
diff --git a/tests/test_python.py b/tests/test_python.py index ee1e1f9..c4e333e 100644 --- a/tests/test_python.py +++ b/tests/test_python.py @@ -5,8 +5,9 @@ import os import sys +from coverage import env -from coverage.python import get_zip_bytes +from coverage.python import get_zip_bytes, source_for_file from tests.coveragetest import CoverageTest @@ -28,3 +29,27 @@ class GetZipBytesTest(CoverageTest): self.assertIn('All OK', zip_text) # Run the code to see that we really got it encoded properly. __import__("encoded_"+encoding) + +def test_source_for_file(tmpdir): + path = tmpdir.join("a.py") + src = str(path) + assert src == source_for_file(src) + assert src == source_for_file(src + 'c') + assert src == source_for_file(src + 'o') + unknown = src + 'FOO' + assert unknown == source_for_file(unknown) + # + # On windows if a pyw exists, it is an acceptable source + # + windows = env.WINDOWS + env.WINDOWS = True + path_windows = tmpdir.ensure("a.pyw") + assert str(path_windows) == source_for_file(src + 'c') + # + # If both pyw and py exist, py is preferred + # + path.ensure(file=True) + assert src == source_for_file(src + 'c') + env.WINDOWS = windows + jython_src = "a" + assert jython_src + ".py" == source_for_file(jython_src + "$py.class") |