diff --git a/nmma/tests/analysis.py b/nmma/tests/analysis.py index 35c6e9d9..1f391025 100644 --- a/nmma/tests/analysis.py +++ b/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): diff --git a/nmma/tests/data/Ka2017_injection.json b/nmma/tests/data/Ka2017_injection.json new file mode 100644 index 00000000..e7fef5b7 --- /dev/null +++ b/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 + ] + } + } +}