Add simple html cleaning function for testing

jieter · Dec 10, 2016 · 2ae0ae6 · 2ae0ae6
1 parent 6fcaaf3
commit 2ae0ae6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/utils.py b/tests/utils.py
@@ -1,3 +1,4 @@
+import re
 import warnings
 from contextlib import contextmanager
 
@@ -78,3 +79,8 @@ def build_request(uri='/'):
         'wsgi.multithread': False,
         'wsgi.run_once': False,
     })
+
+
+def clean_output(s):
+    '''Remove double newlines with whitespace in between and reduce the level of indentation'''
+    return re.sub('\n( *\n)+', '\n', s).replace('    ', '  ')