# -*- 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())