def test_get_id_from_href_exception(self): href = "525/323" with self.assertRaises(InvalidInput): douban_id = get_id_from_href(href)
def test_get_id_from_href(self): href = "http://movie.douban.com/subject/1421721/" douban_id = get_id_from_href(href) self.assertEquals(douban_id, 1421721)