def test_remove_object_raises_error_if_invalid_index(self, sets): manager = ObjectsManager(sets) obj_index = manager.create(self.options) self.assertEqual(len(manager.all()), sets) with self.assertRaises(ValueError): manager.remove(100) self.assertEqual(len(manager.all()), sets)
def test_can_remove_object(self, sets): manager = ObjectsManager(sets) obj_index = manager.create(self.options) self.assertEqual(len(manager.all()), 1 * sets) manager.remove(obj_index) self.assertEqual(len(manager.all()), 0)