diff --git a/pywb/rewrite/test/test_html_rewriter.py b/pywb/rewrite/test/test_html_rewriter.py
index 002529e6..5f279d87 100644
--- a/pywb/rewrite/test/test_html_rewriter.py
+++ b/pywb/rewrite/test/test_html_rewriter.py
@@ -159,6 +159,14 @@ ur"""
>>> parse('Some Text without any tags ', head_insert = '')
Some Text without any tags
+# no parse comments
+>>> parse('')
+
+
+# with parse comments
+>>> parse('', parse_comments=True)
+
+
# rel=canonical: rewrite (default)
>>> parse('')
@@ -236,8 +244,10 @@ urlrewriter_pencode = new_rewriter(rewrite_opts=dict(punycode_links=True))
no_base_canon_rewriter = new_rewriter(rewrite_opts=dict(rewrite_rel_canon=False,
rewrite_base=False))
-def parse(data, head_insert=None, urlrewriter=urlrewriter):
- parser = HTMLRewriter(urlrewriter, head_insert = head_insert, url = ORIGINAL_URL)
+def parse(data, head_insert=None, urlrewriter=urlrewriter, parse_comments=False):
+ parser = HTMLRewriter(urlrewriter, head_insert=head_insert,
+ url=ORIGINAL_URL,
+ parse_comments=parse_comments)
if isinstance(data, unicode):
data = data.encode('utf-8')