aboutsummaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_rst_tables.py22
1 files changed, 17 insertions, 5 deletions
diff --git a/tests/test_rst_tables.py b/tests/test_rst_tables.py
index 0f1f354..381adf8 100644
--- a/tests/test_rst_tables.py
+++ b/tests/test_rst_tables.py
@@ -22,7 +22,8 @@ import unittest
# Load test subjects
from rst_tables import get_table_bounds, reformat_table, parse_table, \
draw_table, table_line, get_column_widths, \
- pad_fields, unify_table, join_rows
+ pad_fields, unify_table, join_rows, \
+ partition_raw_lines
class TestRSTTableFormatter(unittest.TestCase):
@@ -67,6 +68,19 @@ class TestRSTTableFormatter(unittest.TestCase):
expected = ['x foo apple', 'y bar', 'z pear']
self.assertEquals(expected, join_rows(input_rows, sep=' '))
+ def testPartitionRawLines(self):
+ self.assertEquals([], partition_raw_lines([]))
+ self.assertEquals([['']], partition_raw_lines(['']))
+ self.assertEquals(
+ [['foo'], ['bar']],
+ partition_raw_lines(['foo', 'bar']))
+ self.assertEquals(
+ [['foo'], ['bar']],
+ partition_raw_lines(['foo', '+----+', 'bar']))
+ self.assertEquals(
+ [['foo', 'bar'], ['baz']],
+ partition_raw_lines(['+-----+', 'foo', 'bar', '----', 'baz']))
+
def testParseSimpleTable(self):
self.assertEquals([['x y z']], parse_table(['x y z']))
self.assertEquals([['x', 'y z']], parse_table(['x y z']))
@@ -119,8 +133,7 @@ class TestRSTTableFormatter(unittest.TestCase):
'blah | A new line| ',
'+-----+----+']
expect = [['Foo', 'Mu'],
- ['x', 'This became somewhat larger'],
- ['blah', 'A new line']]
+ ['x\nblah', 'This became somewhat larger\nA new line']]
self.assertEquals(expect, parse_table(input))
input = ['+===+-----====+',
@@ -130,8 +143,7 @@ class TestRSTTableFormatter(unittest.TestCase):
'blah | A new line|| ',
'+-----+----+']
expect = [['Foo', 'Mu'],
- ['x', 'This became somewhat larger'],
- ['blah', 'A new line']]
+ ['x\nblah', 'This became somewhat larger\nA new line']]
self.assertEquals(expect, parse_table(input))
def testTableLine(self):