diff --git a/GEODE/Visualisation/DensityProfile.py b/GEODE/Visualisation/DensityProfile.py index 4e8c4bd162fd988c35c70821575e5d322d3d35d5..36745a95566fd32f8267c118964629591463c5d7 100644 --- a/GEODE/Visualisation/DensityProfile.py +++ b/GEODE/Visualisation/DensityProfile.py @@ -42,13 +42,16 @@ def densityProfile(measures, resolution): def drawDensityProfile(measures, outputFile, resolution): plotDensity(densityProfile(measures, resolution), outputFile) -def drawDensityProfileCLI(arguments): +def getArgs(arguments): cli = argparse.ArgumentParser( prog='densityProfile', description="Draw a density profile from a set of occurrences") cli.add_argument('inputTSV') cli.add_argument('outputPNG') cli.add_argument('-r', '--resolution', type=int) - args = cli.parse_args(arguments) + return cli.parse_args(arguments) + +def drawDensityProfileCLI(arguments): + args = getArgs(arguments) measures = [m[1] for m in tabular(args.inputTSV).iterrows()] drawDensityProfile(measures, args.outputPNG, args.resolution or 100)