def test_start_with_watermark(self): 'remove_watermark should remove the watermark' html_orig = _load_fixture(self.page_name, 'has_watermark') removed, html_observed = remove_watermark(html_orig) html_expected =_load_fixture(self.page_name, 'no_watermark') n.assert_true(removed) assert_xml_equal(html_observed, html_expected)
def test_start_without_watermark(self): 'remove_watermark should do nothing if the watermark isn\'t there.' html_expected =_load_fixture(self.page_name, 'no_watermark') removed, html_observed = remove_watermark(html_expected) n.assert_false(removed) assert_xml_equal(html_observed, html_expected)