def test_remove_relations(self): context = Context() context.create_document(1, 'python.pdf', 'pdf', '/tmp/python.pdf') context.create_document(2, 'rust.pdf', 'pdf', '/tmp/rust.pdf') context.create_document(3, 'lua.pdf', 'pdf', '/tmp/lua.pdf') context.create_tag(1, 'book') context.create_relation(1, 1) context.create_relation(2, 1) context.create_relation(3, 1) self.assertEqual(context.count_relations(), 3) context.destroy_relation(2, 1) self.assertEqual(context.count_relations(), 2) context.destroy_relation(1, 1) self.assertEqual(context.count_relations(), 1) context.destroy_relation(3, 1) self.assertEqual(context.count_relations(), 0)
def test_remove_missing_relation(self): context = Context() context.create_tag(1, 'book') context.create_document(1, 'python.pdf', 'pdf', '/tmp/python.pdf') with self.assertRaises(ValueError): context.destroy_relation(1, 1)