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)