def test_single_key_removed(self):
test_data = [
{'a_key': 'a_value'},
]
keys = [
'a_key'
]
p = RemoveKeys(keys)
p.bind(test_data)
for i in p:
self.assertNotIn('a_key', i)
def test_remaining_key_unaffected(self):
test_data = [
{'a_key': 'a_value', 'a_n_other_key': 'a_n_other_value'},
]
keys = [
'a_key'
]
p = RemoveKeys(keys)
p.bind(test_data)
for i in p:
self.assertNotIn('a_key', i)
self.assertIn('a_n_other_key', i)
self.assertEquals(i['a_n_other_key'], 'a_n_other_value')
