def test_accepts_list_of_partial_matches(self): """Test that a list of partial patterns removes the expected attributes.""" attributes_to_delete = ['tit', 'mosg_grid'] expected = {'mosg_model': 'gl_det'} delete_attributes(self.cube, attributes_to_delete) self.assertDictEqual(expected, self.cube.attributes)
def test_accepts_string(self): """Test that a single string passed as an argument works.""" attributes_to_delete = 'title' attributes = copy(self.cube.attributes) attributes.pop(attributes_to_delete) delete_attributes(self.cube, attributes_to_delete) self.assertDictEqual(attributes, self.cube.attributes)
def test_accepts_list_of_complete_matches(self): """Test that a list of complete attribute names removes the expected attributes.""" attributes_to_delete = ['title', 'tithe', 'mosg_model'] attributes = copy(self.cube.attributes) for item in attributes_to_delete: attributes.pop(item) delete_attributes(self.cube, attributes_to_delete) self.assertDictEqual(attributes, self.cube.attributes)
def test_basic(self): """Test that an empty call leaves the cube unchanged.""" cube = self.cube.copy() delete_attributes(cube, []) self.assertDictEqual(self.cube.attributes, cube.attributes)