From 45b6afc30cf687f5719a73a3051ec70c6e3cae51 Mon Sep 17 00:00:00 2001 From: Campbell Barton Date: Wed, 12 Nov 2014 14:22:40 +0100 Subject: [PATCH] bam cli: Accept argv from another script (for testing) --- client/cli/bam.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/client/cli/bam.py b/client/cli/bam.py index fb28103..0e1b27e 100755 --- a/client/cli/bam.py +++ b/client/cli/bam.py @@ -803,11 +803,14 @@ def create_argparse(): return parser -def main(): - import sys +def main(argv=None): - parser = create_argparse() - args = parser.parse_args(sys.argv[1:]) + if argv is None: + import sys + argv = sys.argv + + parser = create_argparse(argv) + args = parser.parse_args() # call subparser callback if not hasattr(args, "func"):