diff --git a/tests/utils_test.py b/tests/utils_test.py index bb217c1b7..f64b759ae 100644 --- a/tests/utils_test.py +++ b/tests/utils_test.py @@ -518,6 +518,12 @@ def test_find_table_simple(self): "B2", utils.TableDirection.table, ) + + table_max_row_max_column = utils.find_table( + values, + "D3", + utils.TableDirection.table, + ) right = utils.find_table( values, "B2", @@ -535,9 +541,18 @@ def test_find_table_simple(self): ["B2", "C2"], ["B3", "C3"], ] + for rowindex, row in enumerate(table_values): self.assertListEqual(row, table[rowindex]) + table_max_row_max_column_values = [ + ["D3", "E3"], + ["D4", "E4"], + ] + + for rowindex, row in enumerate(table_max_row_max_column): + self.assertListEqual(row, table_max_row_max_column_values[rowindex]) + right_values = [ ["B2", "C2"], ] @@ -645,3 +660,16 @@ def test_find_table_last_column_gap(self): for rowindex, row in enumerate(expected_table): self.assertListEqual(row, table[rowindex]) + + def test_find_table_empty_top_left_corner(self): + """Test find table with an empty top left cell and empty adjacent cells""" + + values = [ + ["", "", "C1", ""], + ["", "B2", "C2", ""], + ["", "B3", "C3", ""], + ] + + table = utils.find_table(values, "A1", utils.TableDirection.table) + + self.assertListEqual(table, [], "resulting table should be empty")