def test_set_attribute(self): #Now we'll repeat the test, using the set_attribute method. tag = Tag('img') tag.set_attribute('src', 'http://i.imgur.com/iWHK2.gif') tag.set_attribute('alt', 'Cats are liquid, not solid.') soup = BeautifulSoup(tag.render()) self.assertEqual(soup.img.name, 'img') self.assertEqual(soup.img['src'], 'http://i.imgur.com/iWHK2.gif') self.assertEqual(soup.img['alt'], 'Cats are liquid, not solid.')
def test_invalid_attributes(self): tag = Tag('img') with self.assertRaises(ValueError): tag.set_attribute('src`', 'http://i.imgur.com/iWHK2.gif')