def test_add_plot(self): fig = Figure() name = "test" plt.figure() plt.plot(x, y) fig.add_plot(name) # check if correct latex output is produced self.assertEqual( fig.dumps(), ("\\begin{figure}%\n\\centering%\n\\includegraphics[width=0.8" + "\\textwidth]{Graphs/test.jpg}%\n\\end{figure}"), ) plt.close()
def test_reset(self): fig = Figure() # figure 1 for i in range(1, 3): name = f"test{i}" plt.figure() plt.plot(x, y, label=i) fig.add_plot(name) # check if correct latex output is produced self.assertEqual( fig.dumps(), (f"\\begin{{figure}}%\n\\centering%\n\\includegraphics[width=0.8\\textwidth]" + f"{{Graphs/test{i}.jpg}}%\n\\end{{figure}}"), ) fig.reset(show=False, close=True)
def test_two(self): doc = Document() fig = Figure() with doc.create(Figure()) as fig: with fig.create( SubFigure(width=NoEscape(r"0.48\textwidth"))) as left: name = "left" plt.figure() plt.plot(x, y, label="l") plt.legend() left.add_plot(name, caption=name) plt.close() with fig.create( SubFigure(width=NoEscape(r"0.48\textwidth"))) as right: name = "right" plt.figure() plt.plot(x, y, label="r") plt.legend() right.add_plot(name, caption=name) plt.close() fig.add_caption_label("full", "full", above=False) self.assertEqual( fig.dumps(), ("\\begin{figure}%\n\\begin{subfigure}{0.48\\textwidth}%" + "\n\\caption{left}%\n\\zlabel{fig:left}%\n\\includegraphics[width=\\linewidth]" + "{Graphs/left.jpg}%\n\\end{subfigure}%\n\\begin{subfigure}{0.48\\textwidth}%" + "\n\\caption{right}%\n\\zlabel{fig:right}%\n\\includegraphics[width=" + "\\linewidth]{Graphs/right.jpg}%\n\\end{subfigure}%\n\\caption{full}%\n" + "\\zlabel{fig:full}%\n\\end{figure}"), ) doc.generate_tex("Latex/twosubfigures")