summaryrefslogtreecommitdiff
path: root/test/test_phystokens.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_phystokens.py')
-rw-r--r--test/test_phystokens.py10
1 files changed, 8 insertions, 2 deletions
diff --git a/test/test_phystokens.py b/test/test_phystokens.py
index 03f2a929..64316034 100644
--- a/test/test_phystokens.py
+++ b/test/test_phystokens.py
@@ -19,17 +19,23 @@ HERE = os.path.split(__file__)[0]
class PhysTokensTest(CoverageTest):
"""Tests for Coverage.py's improver tokenizer."""
+ def setUp(self):
+ self.run_in_temp_dir = False
+ super(PhysTokensTest, self).setUp()
+
def check_tokenization(self, source):
"""Tokenize `source`, then put it back together, should be the same."""
tokenized = ""
for line in source_token_lines(source):
text = "".join([t for _,t in line])
tokenized += text + "\n"
+ # source_token_lines doesn't preserve trailing spaces, so trim all that
+ # before comparing.
source = re.sub("(?m)[ \t]+$", "", source)
tokenized = re.sub("(?m)[ \t]+$", "", tokenized)
#if source != tokenized:
- # open(r"c:\foo\0.py", "w").write(source)
- # open(r"c:\foo\1.py", "w").write(tokenized)
+ # open("0.py", "w").write(source)
+ # open("1.py", "w").write(tokenized)
self.assertEqual(source, tokenized)
def check_file_tokenization(self, fname):