TST: fix string decoding issue in Python 3

ellisonbg · Sep 25, 2016 · 762c917 · 762c917
1 parent a885e91
commit 762c917
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/altair/tests/test_api.py b/altair/tests/test_api.py
@@ -72,7 +72,10 @@ def test_savechart_html():
 
     with tempfile.NamedTemporaryFile(suffix='.html') as f:
         chart.savechart(f.name)
-        assert f.read().strip().startswith('<!DOCTYPE html>')
+        content = f.read()
+        if hasattr(content, 'decode'):
+            content = content.decode('utf-8')
+        assert content.strip().startswith('<!DOCTYPE html>')
 
 
 def test_savechart_json():