Commit ed5b0134 authored by Ambra Di Piano's avatar Ambra Di Piano
Browse files

add wrap and test

parent 6464714e

astrort/testing/test_astrort/test_utils/test_wrap.py

+3 −3
Original line number Diff line number Diff line
@@ -9,7 +9,7 @@ 
import pytest

from astrort.utils.wrap import load_yaml_conf



@pytest.mark.rtadeep_configuration

def test_load_yaml_conf(rtadeep_configuration):

    configuration = load_yaml_conf(rtadeep_configuration)

@pytest.mark.astrort_configuration

def test_load_yaml_conf(astrort_configuration):

    configuration = load_yaml_conf(astrort_configuration)

    assert type(configuration) == dict
 
 No newline at end of file

astrort/utils/wrap.py

+10 −2
Original line number Diff line number Diff line
@@ -7,10 +7,18 @@ 
# *****************************************************************************



import yaml

from astrort.cfg.check_configuration import CheckConfiguration

from astrort.utils.utils import seeds_to_string_formatter, get_instrument_fov

from astrort.configure.check_configuration import CheckConfiguration



def load_yaml_conf(yamlfile):

    with open(yamlfile) as f:

        configuration = yaml.load(f, Loader=yaml.FullLoader)

        CheckConfiguration(configuration=configuration)

    return configuration



def configure_simulator(simulator, configuration):

    simulator.model = configuration['model']

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

    simulator.caldb = configuration['prod']

    simulator.irf = configuration['irf']

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