def test_no_links(self): valid_result = [] res = [] for elem in main.extract_images(self.empty_html): res.append(elem) self.assertEqual(res, valid_result)
def test_none(self): valid_result = [] res = [] for elem in main.extract_images(None): res.append(elem) self.assertEqual(res, valid_result)
def test_valid_links(self): valid_result = [u"test.png", u"rwar.png"] res = [] for elem in main.extract_images(self.html_string_valid): res.append(elem) self.assertEqual(res, valid_result)