def test_fix_rows(self):
"""Test to ensure that row merging yields correct results"""
start = [['1', '2', '3', ],
[''],
['abc'],
['4', '5']
]
fixed = join_rows(start)
self.assertEqual(4, len(fixed))
self.assertEqual(start[0][0], fixed[0])
self.assertEqual(start[0][1], fixed[1])
self.assertEqual(" ".join([start[0][-1], start[1][0], start[2][0], start[3][0]]), fixed[2])
self.assertEqual(start[3][1], fixed[3])
def test_fix_rows(self):
"""Test to ensure that row merging yields correct results"""
start = [['1', '2', '3', ],
[''],
['abc'],
['4', '5']
]
fixed = join_rows(start)
self.assertEqual(4, len(fixed))
self.assertEqual(start[0][0], fixed[0])
self.assertEqual(start[0][1], fixed[1])
self.assertEqual(" ".join([start[0][-1], start[1][0], start[2][0], start[3][0]]), fixed[2])
self.assertEqual(start[3][1], fixed[3])
def test_real_world_join_fail(self):
start = [['168772', '1102', '$0.23 TO $0.72', 'HOUR', '1.5%'],
['GROSS', '1.5% '], ['GROSS', '430938']]
fixed = join_rows(start)
self.assertEqual(7, len(fixed))
self.assertEqual(start[0][0], fixed[0])
self.assertEqual(start[0][1], fixed[1])
self.assertEqual(start[0][2], fixed[2])
self.assertEqual(start[0][3], fixed[3])
expected4 = " ".join([start[0][-1], start[1][0]])
self.assertEqual(expected4, fixed[4])
expected5 = " ".join([start[1][1], start[2][0]])
self.assertEqual(expected5, fixed[5])
self.assertEqual(start[2][1], fixed[6])
def test_real_world_join_fail(self):
start = [['168772', '1102', '$0.23 TO $0.72', 'HOUR', '1.5%'],
['GROSS', '1.5% '],
['GROSS', '430938']]
fixed = join_rows(start)
self.assertEqual(7, len(fixed))
self.assertEqual(start[0][0], fixed[0])
self.assertEqual(start[0][1], fixed[1])
self.assertEqual(start[0][2], fixed[2])
self.assertEqual(start[0][3], fixed[3])
expected4 = " ".join([start[0][-1], start[1][0]])
self.assertEqual(expected4, fixed[4])
expected5 = " ".join([start[1][1], start[2][0]])
self.assertEqual(expected5, fixed[5])
self.assertEqual(start[2][1], fixed[6])