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)
