def test_plot_arg(cmd, expected):
"""Test plot args."""
parser = argparse.ArgumentParser()
lc.add_plot_arg(parser)
v = vars(parser.parse_args(cmd))
assert v['plot'] is True
assert 'plotvars' in v
assert v['plotvars'] == expected
_name = 'pangle'
ap = libcli.CustomParser(
description=_BANNER + __doc__,
formatter_class=argparse.RawDescriptionHelpFormatter,
)
libcli.add_version_arg(ap)
libcli.add_topology_arg(ap)
libcli.add_trajectories_arg(ap)
libcli.add_plane_selection_arg(ap)
libcli.add_angle_unit_arg(ap)
libcli.add_reference_frame_arg(ap)
libcli.add_slice_arg(ap)
libcli.add_data_export_arg(ap)
libcli.add_plot_arg(ap)
def _ap():
return ap
def main(topology,
trajectories,
plane_selection,
aunit='degrees',
ref_frame=0,
start=None,
stop=None,
step=None,
export=False,