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)
