def test_formatter_by_named_column(self): """Test one named column""" s = NominableSheet(self.data, "test") s.name_columns_by_row(0) f = NamedColumnFormatter("Column 1", str) s.apply_formatter(f) assert s.column["Column 1"] == ["1", "4", "7"]
def test_series(self): s = NominableSheet(self.data, "test") s.name_columns_by_row(2) assert s.colnames == ["Column 1", "Column 2", "Column 3"] custom_columns = ["C1", "C2", "C3"] s.colnames = custom_columns assert s.colnames == custom_columns
def test_add(self): s = NominableSheet(self.data, "test") s.name_columns_by_row(0) data = OrderedDict({ "Column 4": [10, 11, 12] }) s = s.column + data assert s.column["Column 4"] == [10, 11, 12]
def test_formatter_by_named_columns(self): """Test multiple named columns""" s = NominableSheet(self.data, "test") s.name_columns_by_row(0) f = NamedColumnFormatter(["Column 1", "Column 3"], str) s.apply_formatter(f) assert s.column["Column 1"] == ["1", "4", "7"] assert s.column["Column 3"] == ["3", "6", "9"]
def test_add_wrong_type(self): """Add string type""" s = NominableSheet(self.data, "test") s.name_columns_by_row(0) s = s.column + "string type" # bang
def test_set_indexed_row(self): s = NominableSheet(self.data, "test") s.name_columns_by_row(2) s.row[0] = [10000, 1, 11] assert s.row[0] == [10000, 1, 11]
def test_delete_named_column(self): s = NominableSheet(self.data, "test") s.name_columns_by_row(2) del s.column["Column 2"] assert s.number_of_columns() == 2 s.column["Column 2"] # bang
def test_formatter_by_named_column_2(self): s = NominableSheet(self.data, "test") s.name_columns_by_row(2) s.column.format("Column 1", str) assert s.column["Column 1"] == ["1", "4", "7"]
def test_delete_indexed_column(self): s = NominableSheet(self.data, "test") s.name_columns_by_row(0) s.delete_named_column_at(1) assert s.number_of_columns() == 2 s.column["Column 2"] # access it after deletion, bang