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)