add injection file and injection path

nmma/tests/analysis.py

@@ -7,15 +7,15 @@
 from tools import analysis_slurm
 
 
+WORKING_DIR = os.path.dirname(__file__)
+DATA_DIR = os.path.join(WORKING_DIR, "data")
+
 @pytest.fixture(scope="module")
 def args():
-    workingDir = os.path.dirname(__file__)
-    dataDir = os.path.join(workingDir, "data")
-
     args = Namespace(
         model="Bu2019nsbh",
         interpolation_type="tensorflow",
-        svd_path=dataDir,
+        svd_path=DATA_DIR,
         outdir="outdir",
         label="injection",
         trigger_time=None,
@@ -40,7 +40,7 @@ def args():
         nlive=64,
         reactive_sampling=False,
         seed=42,
-        injection=f"{dataDir}/Bu2019lm_injection.json",
+        injection=f"{DATA_DIR}/Bu2019lm_injection.json",
         injection_num=0,
         injection_detection_limit=None,
         injection_outfile="outdir/lc.csv",
@@ -99,7 +99,7 @@ def test_nn_analysis(args):
     args.dt = 0.25
     args.filters = "ztfg,ztfr,ztfi"
     args.local_only = False
-
+    args.injection=f"{DATA_DIR}/Ka2017_injection.json"
     analysis.main(args)
 
 def test_analysis_slurm(args):

nmma/tests/data/Ka2017_injection.json

@@ -0,0 +1,28 @@
+{
+  "injections": {
+    "__dataframe__": true,
+    "content": {
+      "simulation_id": [
+        0
+      ],
+      "luminosity_distance": [
+        41.521910341458366
+      ],
+      "timeshift": [
+        0.0
+      ],
+      "log10_mej": [
+        -2.1686855651216765
+      ],
+      "log10_vej": [
+        -0.9281116614254382
+      ],
+      "log10_Xlan": [
+        -4.9135946404939235
+      ],
+      "geocent_time": [
+        58940.94264517932
+      ]
+    }
+  }
+}