Feat(tests): Use tox for testing

[mimedecode.git] / test / html2txt.py

diff --git a/test/html2txt.py b/test/html2txt.py
index 4fa49f8cb931067cee9c43fe4f43c949d26c1bed..6b47f77b4d3c80d94645e5ef6bae6feecf4a6718 100755 (executable)
--- a/test/html2txt.py
+++ b/test/html2txt.py
@@ -4,11 +4,21 @@ from __future__ import print_function
 import sys
 from m_lib.net.www.html import HTMLFilter
 
-with open(sys.argv[1], 'r') as f:
-    html = f.read()
+PY2 = sys.version_info[0] < 3
+if PY2:
+    with open(sys.argv[1], 'r') as f:
+        html = f.read()
+else:
+    with open(sys.argv[1], 'r', encoding='utf-8') as f:
+        html = f.read()
 
 filter = HTMLFilter()
 filter.feed(html)
 filter.close()
 
-print(filter.accumulator)
+if PY2:
+    print(filter.accumulator)
+else:
+    if not isinstance(filter.accumulator, bytes):
+        filter.accumulator = filter.accumulator.encode('utf-8')
+    sys.stdout.buffer.write(filter.accumulator + b'\n')