feat: allow create_ntuples utility to accept an output directory from…

… a CLI (#508)

* Allow `create_ntuples` util script to accept output directory from CLI.

tests/test_integration.py

@@ -1,5 +1,6 @@
 import logging
 
+from click.testing import CliRunner
 import numpy as np
 import pytest
 
@@ -22,7 +23,8 @@ def test_integration(tmp_path, ntuple_creator, caplog):
     """The purpose of this integration test is to check whether the
     steps run without error and whether the fit result is as expected.
     """
-    ntuple_creator(str(tmp_path))
+    runner = CliRunner()
+    runner.invoke(ntuple_creator, ["--output_directory", str(tmp_path)])
 
     cabinetry_config = cabinetry.configuration.load("config_example.yml")
 

utils/create_ntuples.py

@@ -2,6 +2,7 @@
 
 import os
 
+import click
 import matplotlib.pyplot as plt
 import numpy as np
 import uproot
@@ -151,7 +152,17 @@ def plot_distributions(data, weights, labels, pseudodata, bins):
     plt.close()
 
 
+@click.command()
+@click.option(
+    "--output_directory",
+    type=str,
+    default="inputs/",
+    help="Output directory for ntuples used in simple example, useful for development.",
+)
 def run(output_directory, *, visualize=False):
+    # Create the output directory
+    if not os.path.exists(output_directory):
+        os.mkdir(output_directory)
     # configuration
     num_events = 5000
     yield_s = 125
@@ -202,7 +213,4 @@ def run(output_directory, *, visualize=False):
 
 
 if __name__ == "__main__":
-    output_directory = "inputs/"
-    if not os.path.exists(output_directory):
-        os.mkdir(output_directory)
-    run(output_directory)
+    run()