aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--tests/fixtures/default.txt9
-rw-r--r--tests/test_rst_tables.py41
2 files changed, 28 insertions, 22 deletions
diff --git a/tests/fixtures/default.txt b/tests/fixtures/default.txt
new file mode 100644
index 0000000..ff19f0e
--- /dev/null
+++ b/tests/fixtures/default.txt
@@ -0,0 +1,9 @@
+This is paragraph text *before* the table.
+
+Column 1 Column 2
+Foo Put two (or more) spaces as a field separator.
+Bar Even very very long lines like these are fine, as long as you do not put in line endings here.
+Qux This is the last line.
+
+This is paragraph text *after* the table, with
+a line ending.
diff --git a/tests/test_rst_tables.py b/tests/test_rst_tables.py
index f7f5280..1eb61a6 100644
--- a/tests/test_rst_tables.py
+++ b/tests/test_rst_tables.py
@@ -15,7 +15,11 @@ vim.eval = fake_eval
vim.current = mock.Mock()
vimvar['foo'] = 'bar'
+# Begin normal module loading
+import os
import unittest
+
+# Load test subjects
from rst_tables import get_table_bounds, create_table, parse_table, \
draw_table, table_line, get_column_widths, \
pad_fields
@@ -23,36 +27,31 @@ from rst_tables import get_table_bounds, create_table, parse_table, \
class TestRSTTableFormatter(unittest.TestCase):
def setUp(self):
- self.plain = """\
-This is paragraph text *before* the table.
+ # Default vim cursor for all tests is at line 4
+ vim.current = mock.Mock()
+ vim.current.window.cursor = (4, 0)
-Column 1 Column 2
-Foo Put two (or more) spaces as a field separator.
-Bar Even very very long lines like these are fine, as long as you do not put in line endings here.
-Qux This is the last line.
+ def tearDown(self):
+ del vim.current
-This is paragraph text *after* the table, with
-a line ending.
-"""
+ def read_fixture(self, name):
+ return open(os.path.join('tests/fixtures/', name + '.txt'),
+ 'r').read().split('\n')
- def tearDown(self):
- pass
+ def load_fixture_in_vim(self, name):
+ vim.current.buffer = self.read_fixture(name)
def testGetBounds(self):
- input = self.plain
- vim.current.buffer = input.split('\n')
- vim.current.window.cursor = (4, 0)
+ self.load_fixture_in_vim('default')
self.assertEquals((3, 6), get_table_bounds())
def testGetBoundsOnBeginOfFile(self):
- input = self.plain
- vim.current.buffer = input.split('\n')
+ self.load_fixture_in_vim('default')
vim.current.window.cursor = (1, 0)
self.assertEquals((1, 1), get_table_bounds())
def testGetBoundsOnEndOfFile(self):
- input = self.plain
- vim.current.buffer = input.split('\n')
+ self.load_fixture_in_vim('default')
vim.current.window.cursor = (8, 0)
self.assertEquals((8, 9), get_table_bounds())
@@ -62,8 +61,7 @@ a line ending.
self.assertEquals([['x', 'y', 'z']], parse_table(['x y z']))
def testParseTable(self):
- input = self.plain
- vim.current.buffer = input.split('\n')
+ self.load_fixture_in_vim('default')
expected = [
['Column 1', 'Column 2'],
['Foo', 'Put two (or more) spaces as a field separator.'],
@@ -161,7 +159,7 @@ a line ending.
draw_table([['Foo', 'Mu'], ['x', 'y']]))
def testCreateTable(self):
- input = self.plain
+ self.load_fixture_in_vim('default')
expect = """\
This is paragraph text *before* the table.
@@ -178,7 +176,6 @@ This is paragraph text *before* the table.
This is paragraph text *after* the table, with
a line ending.
""".split('\n')
- vim.current.buffer = input.split('\n')
vim.current.window.cursor = (4, 10)
create_table()
self.assertEquals(expect, vim.current.buffer)