test_data: preserve leading whitespace
This is required for the pretty printing tests.
diff --git a/test_data/__init__.py b/test_data/__init__.py
index dbea7e0..f21770b 100644
--- a/test_data/__init__.py
+++ b/test_data/__init__.py
@@ -61,7 +61,7 @@
with open(os.path.join(_test_data_dir, name)) as f:
for line in f:
- line = line.strip().partition('#')[0].strip()
+ line = line.rstrip().partition('#')[0].rstrip()
if line == '':
continue
elif line.startswith('--'):
diff --git a/test_data/example.data b/test_data/example.data
index ea2a0ae..d6186e8 100644
--- a/test_data/example.data
+++ b/test_data/example.data
@@ -4,7 +4,7 @@
# That was a blank line
This is not a blank line
-- section1
-abc def
+ abc def
ghi
-- section2
123
diff --git a/utest/test_test_data.py b/utest/test_test_data.py
index 6a4a8ff..16cf440 100755
--- a/utest/test_test_data.py
+++ b/utest/test_test_data.py
@@ -34,7 +34,7 @@
self.assertTrue('./example.data' in test_data.list_files())
data = test_data.read('example.data')
self.assertEquals(sorted(['section1', 'section2', 'binary']), sorted(data.keys()))
- self.assertEquals('abc def\nghi', data['section1'])
+ self.assertEquals(' abc def\nghi', data['section1'])
self.assertEquals('123\n456\n789', data['section2'])
self.assertEquals('\x00\x01\x02\x03\x04\x05\x06\x07\x77\x66\x55\x44\x33\x22\x11\x00',
data['binary'])