red-mail/redmail/test/helpers/resources.py

import numpy as np
from io import BytesIO

import pytest

def get_mpl_fig():
    pytest.importorskip("matplotlib")
    import matplotlib.pyplot as plt
    # Data for plotting
    t = np.arange(0.0, 2.0, 0.01)
    s = 1 + np.sin(2 * np.pi * t)
    fig, ax = plt.subplots()
    ax.plot(t, s)

    buf = BytesIO()
    fig.savefig(buf, format='png')
    buf.seek(0)
    bytes = buf.read()
    return fig, bytes

def get_pil_image():
    pytest.importorskip("PIL")
    from PIL import Image
    img = Image.new('RGB', (100, 30), color = (73, 109, 137))
    buf = BytesIO()
    img.save(buf, format='PNG')
    buf.seek(0)
    bytes = buf.read()
    return img, bytes