From: Mathieu Bridon <bochecha(a)daitauha.fr>
This ensures that we can read/write lines in the new format, as well as
mix lines in both format within a single sources file.
---
test/test_sources.py | 11 ++++++++++-
1 file changed, 10 insertions(+), 1 deletion(-)
diff --git a/test/test_sources.py b/test/test_sources.py
index edff838..c0722b1 100644
--- a/test/test_sources.py
+++ b/test/test_sources.py
@@ -25,6 +25,7 @@ class formatLineTestCase(unittest.TestCase):
def test_correct_entry(self):
CORRECT_ENTRIES = [
(['foo', 'bar'], ('foo bar')),
+ (['foo', 'bar', 'baz'], ('foo bar baz')),
]
for entry, line in CORRECT_ENTRIES:
self.assertEqual(line,
@@ -53,7 +54,8 @@ class parseLineTestCase(unittest.TestCase):
def test_correct_line(self):
CORRECT_LINES = [
('foo bar\n', ['foo', 'bar']),
- ('foo bar\n', ['foo', ' bar'])
+ ('foo bar\n', ['foo', ' bar']),
+ ('foo bar baz\n', ['foo', ' bar', 'baz'])
]
for line, entry in CORRECT_LINES:
self.assertEqual(entry, sources._parse_line(line))
@@ -80,6 +82,10 @@ class ReaderTestCase(unittest.TestCase):
('foo bar\nfooo baaar\n', [['foo', 'bar'],
['fooo', 'baaar'],
]),
+ ('foo bar baz\n', [['foo', 'bar',
'baz']]),
+ ('foo bar\nfoo bar baz\n', [['foo', 'bar'],
+ ['foo', 'bar',
'baz'],
+ ]),
]
for buffer, entries in CORRECT_SOURCES:
fp = StringIO.StringIO(buffer)
@@ -95,6 +101,9 @@ class WriterTestCase(unittest.TestCase):
([['foo', 'bar'],
['fooo', 'baaar'],
], 'foo bar\nfooo baaar\n'),
+ ([['foo', 'bar', 'baz']], 'foo bar
baz\n'),
+ ([['foo', 'bar'], ['foo', 'bar',
'baz']],
+ 'foo bar\nfoo bar baz\n'),
]
for entries, buffer in CORRECT_SOURCES:
fp = StringIO.StringIO()
--
2.1.0