# -*- coding: utf-8 -*-
#
from helpers import assert_equality
def plot():
from matplotlib import pyplot as plt
import numpy as np
fig = plt.figure()
with plt.style.context(("fivethirtyeight")):
np.random.seed(123)
plt.scatter(
np.linspace(0, 100, 101),
np.linspace(0, 100, 101) + 15 * np.random.rand(101),
)
return fig
def test():
assert_equality(plot, __file__[:-3] + "_reference.tex")
return
if __name__ == "__main__":
import helpers
helpers.compare_mpl_latex(plot)
# helpers.print_tree(plot())
def plot():
# plot data
fig = plt.figure()
ax = fig.add_subplot(111)
x = np.arange(3)
y1 = [1, 2, 3]
y2 = [3, 2, 4]
y3 = [5, 3, 1]
w = 0.25
ax.bar(x - w, y1, w, color="b", align="center", label="Data 1")
ax.bar(x, y2, w, color="g", align="center", label="Data 2")
ax.bar(x + w, y3, w, color="r", align="center", label="Data 3")
ax.legend()
return fig
def test():
assert_equality(plot, "test_barchart_legend_reference.tex")
return
if __name__ == "__main__":
import helpers
helpers.compare_mpl_latex(plot)
# helpers.print_tree(plot())
