def test_remove(self): new_attr_set = AttributeSet({self._user_agent, self._timezone}) self.assertEqual(2, len(new_attr_set)) self.assertIn(self._user_agent, new_attr_set) self.assertIn(self._timezone, new_attr_set) new_attr_set.remove(self._user_agent) self.assertEqual(1, len(new_attr_set)) self.assertIn(self._timezone, new_attr_set) self.assertNotIn(self._user_agent, new_attr_set)
def test_remove_attribute_not_present(self): new_attr_set = AttributeSet({self._user_agent, self._timezone}) self.assertEqual(2, len(new_attr_set)) self.assertIn(self._user_agent, new_attr_set) self.assertIn(self._timezone, new_attr_set) non_present_attribute = Attribute(42, 'unknown') with self.assertRaises(KeyError): new_attr_set.remove(non_present_attribute) self.assertEqual(2, len(new_attr_set)) self.assertIn(self._user_agent, new_attr_set) self.assertIn(self._timezone, new_attr_set)