def test_add_image_html_yfrogpic(self):
"""Test replacing yfrog pic text with a link to yfrog pic."""
tweet = "Some text #ahashtag http://yfrog.com/b9hmwkj some more text."
converted_tweet = 'Some text #ahashtag http://yfrog.com/b9hmwkj some more text. </br> <a href="http://yfrog.com/b9hmwkj"><img src="http://yfrog.com/b9hmwkj.th.jpg"/></a>'
self.assertEqual(_add_image_html(tweet), converted_tweet)
def test_add_image_html_twitpic(self):
"""Test replacing twitpic text with a link to twitpic."""
tweet = "Some text #ahashtag http://twitpic.com/354bkr some more text."
converted_tweet = 'Some text #ahashtag http://twitpic.com/354bkr some more text. </br> <a href="http://twitpic.com/354bkr"><img src="http://twitpic.com/show/mini/354bkr"/></a>'
self.assertEqual(_add_image_html(tweet), converted_tweet)
