asgardpy.data.tests.test_target.test_set_models()

Complexity

Conditions

3

Size

Total Lines	74
Code Lines	53

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

import pytest
def test_set_models(base_config, gammapy_data_path):
    """Test non-standard components of Target module."""

    from asgardpy.analysis import AsgardpyAnalysis
    from asgardpy.data.target import set_models

    ebl_file_name = "ebl_franceschini_2017.fits.gz"
    ebl_file = f"{gammapy_data_path}ebl/{ebl_file_name}"
    model_file_0 = f"{gammapy_data_path}fermi-3fhl-crab/Fermi-LAT-3FHL_models.yaml"
    model_file_1 = f"{gammapy_data_path}fermi-3fhl-crab/Fermi-LAT-3FHL_datasets.yaml"

    base_config.target.components[0].spectral.ebl_abs.filename = ebl_file

    analysis_0 = AsgardpyAnalysis(base_config)
    analysis_1 = AsgardpyAnalysis(base_config)
    analysis_2 = AsgardpyAnalysis(base_config)

    # Check when using create_source_skymodel function
    analysis_0.config.target.from_3d = True
    analysis_2.config.target.from_3d = True

    analysis_0.run(["datasets-3d"])
    analysis_2.run(["datasets-3d"])
    # Check when using read_models_from_asgardpy_config
    analysis_1.run(["datasets-1d"])

    analysis_1.config.target.source_name = "4FGL J0534.5+2201i"
    analysis_0.config.target.source_name = "Crab Nebula"

    data_0, model_0 = set_models(
        analysis_0.config.target,
        analysis_0.datasets,
        datasets_name_list=None,
        models=model_file_0,
    )

    data_1, model_1 = set_models(
        analysis_0.config.target,
        analysis_0.datasets,
        datasets_name_list=None,
    )
    data_2, model_2 = set_models(
        analysis_1.config.target,
        analysis_1.datasets,
        datasets_name_list=None,
    )

    # Check when not providing target source name for creating the center of ROI in the exclusion mask
    analysis_2.config.target.source_name = ""
    data_3, model_3 = set_models(
        analysis_2.config.target,
        analysis_2.datasets,
        datasets_name_list=None,
    )

    with pytest.raises(KeyError):
        _, _ = set_models(
            analysis_0.config.target,
            analysis_0.datasets,
            datasets_name_list=None,
            models=model_file_1,
        )
    with pytest.raises(TypeError):
        _, _ = set_models(
            analysis_0.config.target,
            analysis_0.datasets,
            datasets_name_list=None,
            models=1,
        )
    assert model_0[0].datasets_names == ["Fermi-LAT_00", "Fermi-LAT_01"]
    assert model_1[0].spectral_model.model2.filename.name == ebl_file_name
    assert model_2[0].spectral_model.model2.filename.name == ebl_file_name
    assert analysis_0.final_model[0].spectral_model.model2.filename.name == ebl_file_name
    assert model_3[0].name == ""