def test_add_figure_to_parser(self): kah.add_figure_to_parser( self.parser, default_figname="figure", add_limits=True ) args = kah.get_args(self.parser, ["--xmin", "0.5"]) self.assertEqual(args.figname, "figure") self.assertEqual(args.fig_extension, "png") self.assertEqual(args.xmin, 0.5)
from kuibit.simdir import SimDir from kuibit import argparse_helper as pah from kuibit.visualize_matplotlib import ( setup_matplotlib, add_text_to_figure_corner, save, ) """This script plots the trajectories of given apparent horizons. """ if __name__ == "__main__": setup_matplotlib() desc = """Plot trajectories of given apparent horizons.""" parser = pah.init_argparse(desc) pah.add_figure_to_parser(parser) parser.add_argument( "-t", "--type", type=str, choices=["3D", "xy", "xz", "yz"], default="3D", help="Type of plot: 3D, or of a specific plane (default: %(default)s).", ) parser.add_argument( "-a", "--horizons", type=int, required=True,
get_figname, save_from_dir_filename_ext, set_axis_limits_from_args, setup_matplotlib, ) if __name__ == "__main__": setup_matplotlib() desc = f"""\ {kah.get_program_name()} plots the instantaneous and cumulative linear momentum lost via emission of gravitational-wave as a function of time for a given detector. """ parser = kah.init_argparse(desc) kah.add_figure_to_parser(parser, add_limits=True) parser.add_argument( "--detector-num", type=int, required=True, help="Number of the spherical surface over which to read Psi4.", ) parser.add_argument( "--pcut", type=int, required=True, help="Period that enters the fixed-frequency integration." " Typically, the longest physical period in the signal.", )