def test_find_tag_id(self): context = Context() context.create_tag(1, 'python') context.create_tag(2, 'rust') context.create_tag(3, 'lua') self.assertEqual(context.find_tag_id('rust'), 2) self.assertEqual(context.find_tag_id('lua'), 3) self.assertEqual(context.find_tag_id('python'), 1)
def test_find_id_of_missing_tag(self): context = Context() context.create_tag(1, 'python') context.create_tag(2, 'rust') context.create_tag(3, 'lua') with self.assertRaises(ValueError): _ = context.find_tag_id('java')