summaryrefslogtreecommitdiff
path: root/tests/test_oddball.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_oddball.py')
-rw-r--r--tests/test_oddball.py9
1 files changed, 4 insertions, 5 deletions
diff --git a/tests/test_oddball.py b/tests/test_oddball.py
index f136eb49..3352a0cf 100644
--- a/tests/test_oddball.py
+++ b/tests/test_oddball.py
@@ -308,11 +308,10 @@ class ExceptionTest(CoverageTest):
for f, llist in lines.items():
# f is a path to a python module, so we drop the '.py' to get
# a callname
- callname = os.path.basename(f)[:-3]
- if callname not in callnames:
- # ignore this file.
- continue
- clean_lines[os.path.basename(f)] = llist
+ basename = os.path.basename(f)
+ assert basename.endswith(".py")
+ if basename[:-3] in callnames:
+ clean_lines[basename] = llist
self.assertEqual(clean_lines, lines_expected)