Take argument-handling out of the CLI function to draw DensityProfiles

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)