def test_get_args(self):
sys.argv = ["program_name", "--datadir", "test"]
# Test with args = None
args = kah.get_args(self.parser)
self.assertEqual(args.datadir, "test")
# Tests with non None args
args2 = kah.get_args(self.parser, ["--outdir", "test2"])
self.assertEqual(args2.outdir, "test2")
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)
def test_add_grid_structure_to_parser(self):
kah.add_grid_structure_to_parser(self.parser,
edge_color="r",
alpha=0.5)
# The [] essentially means "use defaults"
args = kah.get_args(self.parser, [])
self.assertFalse(args.rl_show)
self.assertEqual(args.rl_edge_color, "r")
self.assertEqual(args.rl_alpha, 0.5)
def test_add_horizon_to_parser(self):
kah.add_horizon_to_parser(
self.parser, color="w", edge_color="r", alpha=0.5, time_tolerance=1
)
# The [] essentially means "use defaults"
args = kah.get_args(self.parser, [])
self.assertFalse(args.ah_show)
self.assertEqual(args.ah_color, "w")
self.assertEqual(args.ah_edge_color, "r")
self.assertEqual(args.ah_alpha, 0.5)
self.assertEqual(args.ah_time_tolerance, 1)
def test_add_grid_to_parser(self):
# Wrong dimensions
with self.assertRaises(ValueError):
kah.add_grid_to_parser(self.parser, dimensions=5)
parser_3D = kah.init_argparse()
kah.add_grid_to_parser(parser_3D, dimensions=3)
# The [] essentially means "use defaults"
args = kah.get_args(parser_3D, [])
self.assertEqual(args.resolution, 500)
self.assertCountEqual(args.origin, [0, 0, 0])
self.assertCountEqual(args.corner, [1, 1, 1])
# 2D
parser_2D = kah.init_argparse()
kah.add_grid_to_parser(parser_2D, dimensions=2)
# The [] essentially means "use defaults"
args = kah.get_args(parser_2D, [])
self.assertEqual(args.plane, "xy")
self.assertCountEqual(args.origin, [0, 0])
self.assertCountEqual(args.corner, [1, 1])
# 1D
parser_1D = kah.init_argparse()
kah.add_grid_to_parser(parser_1D, dimensions=1)
# The [] essentially means "use defaults"
args = kah.get_args(parser_1D, [])
self.assertEqual(args.axis, "x")
self.assertCountEqual(args.origin, [0])
self.assertCountEqual(args.corner, [1])
type=int,
help="Index of the horizon according to QuasiLocalMeasures.",
)
parser.add_argument(
"--time",
type=float,
default=0,
help="Time to consider.",
)
parser.add_argument(
"--estimate-gamma",
action="store_true",
help=("Estimate the Lorentz factor using qlm_w_momentum. "
"Ignore this if you do not know the details"),
)
args = kah.get_args(parser)
logger = logging.getLogger(__name__)
if args.verbose:
logging.basicConfig(format="%(asctime)s - %(message)s")
logger.setLevel(logging.DEBUG)
with SimDir(
args.datadir,
ignore_symlinks=args.ignore_symlinks,
pickle_file=args.pickle_file,
) as sim:
sim_hor = sim.horizons
help="Type of plot: 3D, or of a specific plane (default: %(default)s).",
)
parser.add_argument(
"-a",
"--horizons",
type=int,
required=True,
help="Apparent horizons to plot",
nargs="+",
)
parser.add_argument("--draw-horizons",
help="Draw the apparent horizons, if available",
action="store_true")
args = pah.get_args(parser)
# Parse arguments
logger = logging.getLogger(__name__)
if args.verbose:
logging.basicConfig(format="%(asctime)s - %(message)s")
logger.setLevel(logging.DEBUG)
if args.figname is None:
horizons = "_".join([str(h) for h in args.horizons])
figname = f"ah_{horizons}_trajectory_{args.type}"
else:
figname = args.figname
