class TestColumnManagement(unittest.TestCase): def setUp(self): self.container = TableColumnController(etree.XML(TABLECOLUMNS_C10)) for col in range(len(self.container)): self.container[col] = setdata('checkmark%d' % col) def test_append_one_column(self): self.container.append(1) self.assertEqual(11, len(self.container), "expected 11 columns") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_append_two_columns(self): self.container.append(2) self.assertEqual(12, len(self.container), "expected 12 columns") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_append_count_zero_error(self): with self.assertRaises(ValueError): self.container.append(0) def test_append_negative_count_error(self): with self.assertRaises(ValueError): self.container.append(-1) def test_insert_one_column(self): self.container.insert(5, count=1) self.chk_insert_one_column() def test_insert_one_column_neg_index(self): self.container.insert(-5, count=1) self.chk_insert_one_column() def chk_insert_one_column(self): self.assertEqual(11, len(self.container), "expected 11 columns") self.assertEqual('checkmark4', getdata(self.container[4]), "expected checkmark4 in col 4") self.assertIsNone(getdata(self.container[5]), "column 5 is not None") self.assertEqual('checkmark5', getdata(self.container[6]), "expected checkmark5 in col 6") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_insert_two_columns(self): self.container.insert(5, count=2) self.chk_insert_two_columns() def test_insert_two_columns_neg_index(self): self.container.insert(-5, count=2) self.chk_insert_two_columns() def chk_insert_two_columns(self): self.assertEqual(12, len(self.container), "expected 12 columns") self.assertEqual('checkmark4', getdata(self.container[4]), "expected checkmark4 in col 4") self.assertIsNone(getdata(self.container[5]), "column 5 is not None") self.assertIsNone(getdata(self.container[6]), "column 6 is not None") self.assertEqual('checkmark5', getdata(self.container[7]), "expected checkmark5 in col 7") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_insert_zero_cols_value_error(self): with self.assertRaises(ValueError): self.container.insert(0, count=0) def test_insert_negative_cols_value_error(self): with self.assertRaises(ValueError): self.container.insert(0, count=-1) def test_delete_one_column(self): self.container.delete(5, count=1) self.chk_delete_one_column() def test_delete_one_column_neg_index(self): self.container.delete(-5, count=1) self.chk_delete_one_column() def chk_delete_one_column(self): self.assertEqual(9, len(self.container), "expected 9 columns") self.assertEqual('checkmark4', getdata(self.container[4]), "expected checkmark4 in col 4") self.assertEqual('checkmark6', getdata(self.container[5]), "expected checkmark6 in col 5") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_delete_two_columns(self): self.container.delete(5, count=2) self.chk_delete_two_columns() def test_delete_two_columns_neg_index(self): self.container.delete(-5, count=2) self.chk_delete_two_columns() def chk_delete_two_columns(self): self.assertEqual(8, len(self.container), "expected 8 columns") self.assertEqual('checkmark4', getdata(self.container[4]), "expected checkmark4 in col 4") self.assertEqual('checkmark7', getdata(self.container[5]), "expected checkmark7 in col 5") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_delete_last_column(self): self.container.delete(index=9) self.chk_delete_last_column() def test_delete_last_column_neg_index(self): self.container.delete(index=-1) self.chk_delete_last_column() def chk_delete_last_column(self): self.assertEqual(9, len(self.container), "expected 9 columns") self.assertTrue(self.container.is_consistent()) def test_delete_zero_cols_value_error(self): with self.assertRaises(ValueError): self.container.delete(0, count=0) def test_delete_negative_cols_value_error(self): with self.assertRaises(ValueError): self.container.delete(0, count=-1) def test_delete_cols_index_out_of_range_error(self): with self.assertRaises(IndexError): self.container.delete(10, count=1) def test_delete_cols_index_and_count_out_of_range_error(self): with self.assertRaises(IndexError): self.container.delete(9, count=2)
class TestColumnManagement(unittest.TestCase): def setUp(self): self.container = TableColumnController(etree.XML(TABLECOLUMNS_C10)) for col in range(len(self.container)): self.container[col] = setdata('checkmark%d' % col) def test_append_one_column(self): self.container.append(1) self.assertEqual(11, len(self.container), "expected 11 columns") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_append_two_columns(self): self.container.append(2) self.assertEqual(12, len(self.container), "expected 12 columns") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_append_count_zero_error(self): with self.assertRaises(ValueError): self.container.append(0) def test_append_negative_count_error(self): with self.assertRaises(ValueError): self.container.append(-1) def test_insert_one_column(self): self.container.insert(5, count=1) self.chk_insert_one_column() def test_insert_one_column_neg_index(self): self.container.insert(-5, count=1) self.chk_insert_one_column() def chk_insert_one_column(self): self.assertEqual(11, len(self.container), "expected 11 columns") self.assertEqual('checkmark4', getdata(self.container[4]), "expected checkmark4 in col 4") self.assertIsNone(getdata(self.container[5]), "column 5 is not None") self.assertEqual('checkmark5', getdata(self.container[6]), "expected checkmark5 in col 6") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_insert_two_columns(self): self.container.insert(5, count=2) self.chk_insert_two_columns() def test_insert_two_columns_neg_index(self): self.container.insert(-5, count=2) self.chk_insert_two_columns() def chk_insert_two_columns(self): self.assertEqual(12, len(self.container), "expected 12 columns") self.assertEqual('checkmark4', getdata(self.container[4]), "expected checkmark4 in col 4") self.assertIsNone(getdata(self.container[5]), "column 5 is not None") self.assertIsNone(getdata(self.container[6]), "column 6 is not None") self.assertEqual('checkmark5',getdata(self.container[7]), "expected checkmark5 in col 7") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_insert_zero_cols_value_error(self): with self.assertRaises(ValueError): self.container.insert(0, count=0) def test_insert_negative_cols_value_error(self): with self.assertRaises(ValueError): self.container.insert(0, count=-1) def test_delete_one_column(self): self.container.delete(5, count=1) self.chk_delete_one_column() def test_delete_one_column_neg_index(self): self.container.delete(-5, count=1) self.chk_delete_one_column() def chk_delete_one_column(self): self.assertEqual(9, len(self.container), "expected 9 columns") self.assertEqual('checkmark4' ,getdata(self.container[4]), "expected checkmark4 in col 4") self.assertEqual('checkmark6', getdata(self.container[5]), "expected checkmark6 in col 5") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_delete_two_columns(self): self.container.delete(5, count=2) self.chk_delete_two_columns() def test_delete_two_columns_neg_index(self): self.container.delete(-5, count=2) self.chk_delete_two_columns() def chk_delete_two_columns(self): self.assertEqual(8, len(self.container), "expected 8 columns") self.assertEqual('checkmark4', getdata(self.container[4]), "expected checkmark4 in col 4") self.assertEqual('checkmark7', getdata(self.container[5]), "expected checkmark7 in col 5") self.assertTrue(self.container.is_consistent(), "container structure is not consistant") def test_delete_last_column(self): self.container.delete(index=9) self.chk_delete_last_column() def test_delete_last_column_neg_index(self): self.container.delete(index=-1) self.chk_delete_last_column() def chk_delete_last_column(self): self.assertEqual(9, len(self.container), "expected 9 columns") self.assertTrue(self.container.is_consistent()) def test_delete_zero_cols_value_error(self): with self.assertRaises(ValueError): self.container.delete(0, count=0) def test_delete_negative_cols_value_error(self): with self.assertRaises(ValueError): self.container.delete(0, count=-1) def test_delete_cols_index_out_of_range_error(self): with self.assertRaises(IndexError): self.container.delete(10, count=1) def test_delete_cols_index_and_count_out_of_range_error(self): with self.assertRaises(IndexError): self.container.delete(9, count=2)