pyodide/test/test_matplotlib.py

22 lines
733 B
Python
Raw Normal View History

2018-05-17 15:57:39 +00:00
def test_matplotlib(selenium):
selenium.load_package("matplotlib")
2018-05-17 18:30:52 +00:00
selenium.run("from matplotlib import pyplot as plt")
selenium.run("x = plt.plot([1,2,3])")
2018-05-17 15:57:39 +00:00
img = selenium.run(
"plt.gcf()._repr_html_().src"
)
assert img.startswith('data:image/png;base64,')
2018-05-17 18:30:52 +00:00
assert len(img) == 26766
2018-05-21 18:15:50 +00:00
def test_svg(selenium):
selenium.load_package("matplotlib")
selenium.run("from matplotlib import pyplot as plt")
selenium.run("x = plt.plot([1,2,3])")
selenium.run("import io")
selenium.run("fd = io.BytesIO()")
selenium.run("plt.savefig(fd, format='svg')")
content = selenium.run("fd.getvalue().decode('utf8')")
assert len(content) == 15752
assert content.startswith("<?xml")