def test_that_get_non_empty_rows_returns_non_empty_rows(self):
table_model = TableModel()
empty_row = RowEntries()
table_model.append_table_entry(empty_row)
table_model.append_table_entry(RowEntries(sample_scatter=123))
table_model.append_table_entry(empty_row)
table_model.append_table_entry(RowEntries(sample_direct=345))
table_model.append_table_entry(empty_row)
self.assertEqual(2, len(table_model.get_non_empty_rows()))
for i in table_model.get_non_empty_rows():
self.assertFalse(i.is_empty())
self.assertEqual(5, table_model.get_number_of_rows())
def test_that_get_non_empty_rows_returns_non_empty_rows(self):
table_model = TableModel()
table_index_model = TableIndexModel("", "", "", "", "", "", "",
"", "", "", "", "", "")
table_model.add_table_entry(0, table_index_model)
table_index_model = TableIndexModel('0', "", "", "", "", "", "",
"", "", "", "", "", "")
table_model.add_table_entry(1, table_index_model)
table_index_model = TableIndexModel('', "", "", "", "", "", "",
"", "", "", "5", "", "")
table_model.add_table_entry(2, table_index_model)
table_index_model = TableIndexModel("", "", "", "", "", "", "",
"", "", "", "", "", "")
table_model.add_table_entry(3, table_index_model)
non_empty_rows_actual = table_model.get_non_empty_rows([0, 1, 2, 3])
non_empty_rows_expected = [1, 2]
self.assertEqual(non_empty_rows_actual, non_empty_rows_expected)
