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