Commit 5886b99f authored by Ambra Di Piano's avatar Ambra Di Piano
Browse files

set_irf function

parent 0c53e34b

astrort/testing/test_utils/test_wrap.py

+8 −1
Original line number Original line Diff line number Diff line
@@ -167,3 +167,10 @@ def test_plot_map(test_conf_file, save): 
    if save == 'fits':

    if save == 'fits':

        plotmap = plot_map(fitsmap, log)

        plotmap = plot_map(fitsmap, log)

        assert isfile(plotmap)

        assert isfile(plotmap)



def test_set_irf():

    conf = {'array': 'lst', 'prod': 'prod5-v0.1', 'irf': 'random'}

    log = set_logger(logging.CRITICAL)

    irf = set_irf(conf, log)

    print(irf)

    assert conf['array'].upper() in irf
 
 No newline at end of file

astrort/utils/wrap.py

+8 −6
Original line number Original line Diff line number Diff line
@@ -42,11 +42,6 @@ def configure_simulator_no_visibility(simulator, configuration, log): 
    simulator.model = configuration['model']

    simulator.model = configuration['model']

    simulator.output = seeds_to_string_formatter_files(configuration['samples'], configuration['output'], configuration['name'], configuration['seed'], 'fits')

    simulator.output = seeds_to_string_formatter_files(configuration['samples'], configuration['output'], configuration['name'], configuration['seed'], 'fits')

    simulator.caldb = configuration['prod']

    simulator.caldb = configuration['prod']

    if configuration['irf'] == 'random':

        simulator.irf = select_random_irf(configuration['array'], configuration['prod'])

        log.info(f"Randomising instrument response function [{simulator.irf}]")

    else:

        simulator.irf = configuration['irf']

    simulator.fov = get_instrument_fov(configuration['array'])

    simulator.fov = get_instrument_fov(configuration['array'])

    simulator.t = [0, configuration['duration']]

    simulator.t = [0, configuration['duration']]

    simulator.e = adjust_tev_range_to_irf(get_instrument_tev_range(configuration['array']), simulator.irf)

    simulator.e = adjust_tev_range_to_irf(get_instrument_tev_range(configuration['array']), simulator.irf)
@@ -173,3 +168,10 @@ def plot_map(fitsmap, log): 
    del plot

    del plot

    return plotmap

    return plotmap





def set_irf(configuration, log):

    if configuration['irf'] == 'random':

        irf = select_random_irf(configuration['array'], configuration['prod'])

        log.info(f"Randomising instrument response function [{irf}]")

    else:

        irf = configuration['irf']

    return irf
 
 No newline at end of file