def test_save(self):
x = np.linspace(0, 1, 100)
y = np.sin(x)
plt.plot(x, y)
# Test matplotlib
viz.save("test.pdf")
self.assertTrue(os.path.exists("test.pdf"))
os.remove("test.pdf")
# Test tikzplotlib
viz.save("test.tikz", include_disclaimer=False)
self.assertTrue(os.path.exists("test.tikz"))
# We save the file size so that we can compare it to the "cleaned"
# version
file_size_og = os.stat("test.tikz").st_size
os.remove("test.tikz")
# Test with clean figure
viz.save(
"test.tikz",
tikz_clean_figure=True,
target_resolution=200,
include_disclaimer=False,
)
self.assertTrue(os.path.exists("test.tikz"))
# The cleaned file should be smaller
file_size_cleaned = os.stat("test.tikz").st_size
self.assertLess(file_size_cleaned, file_size_og)
os.remove("test.tikz")
# Try to draw the shape of the horizon
if args.draw_horizons and time in horizons[ah].shape_times:
cut = {
"xy": (None, None, 0),
"xz": (None, 0, None),
"yz": (0, None, None),
}
logger.debug(f"Drawing shape at time {time} for ah {ah}")
shape = horizons[ah].shape_outline_at_time(
time, cut[args.type])
ax.fill(shape[0], shape[1])
else:
logger.debug(f"Shape not available at {time} for ah {ah}")
ax.set_xlabel(to_plot_x)
ax.set_ylabel(to_plot_y)
ax.set_aspect("equal")
ax.legend()
time = x_coord[ah].tmax
add_text_to_figure_corner(fr"$t = {time:.3f}$")
output_path = os.path.join(args.outdir, figname)
logger.debug(f"Saving in {output_path}")
plt.tight_layout()
save(output_path, args.fig_extension, as_tikz=args.as_tikz)