class TestReadmeReader(TestCase): def set_up(self): self.reader = ReadmeReader(text) def test_get_images(self): self.assert_equal(self.reader.get_images(), [ 'http://example.com/image_1.svg', 'http://example.com/image_2.svg', 'http://example.com/screenshot.jpg', 'http://example.com/footnote.jpg', ]) def test_get_screenshot(self): self.assert_equal(self.reader.get_screenshot(), 'http://example.com/screenshot.jpg')
def set_up(self): self.reader = ReadmeReader(text)