Commit a0c8e03d authored by Samuel GAIST's avatar Samuel GAIST
Browse files

[test][plotter] Fix type of comparison parameter

parent f0af08d5
...@@ -29,6 +29,7 @@ ...@@ -29,6 +29,7 @@
import imghdr import imghdr
import numpy import numpy
import six
import import
...@@ -97,7 +98,11 @@ def test_plot_jpeg(): ...@@ -97,7 +98,11 @@ def test_plot_jpeg():
def test_plot_pdf(): def test_plot_pdf():
fig = do_plot('application/pdf') fig = do_plot('application/pdf')
assert fig.startswith('%PDF')
if six.PY2:
assert fig.startswith('%PDF')
assert fig.startswith(b'%PDF')
#with open('test.pdf', 'wb') as f: f.write(fig) #with open('test.pdf', 'wb') as f: f.write(fig)
def test_plot_many_lines(): def test_plot_many_lines():
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment