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.",
)
