asgardpy.config.tests.test_config.test_write_model_config() - Code Metrics - Inspection of "Merge pull request #184 from chaimain/add_lst1_dat..." - chaimain/asgardpy - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Push — main ( 4744cf...d735dd )

by Chaitanya

created 2024-10-17 21:04 UTC

test_write_model_config() A

↳ Parent: asgardpy.config.tests.test_config

Complexity

Conditions

Size

Total Lines	32
Code Lines	21

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	2
eloc	21
nop	0
dl	0
loc	32
rs	9.376
c	0
b	0
f	0

import pytest

from asgardpy.config import AsgardpyConfig


def test_config_basic():
    """Test on basic Config features."""

    config = AsgardpyConfig()
    assert "AsgardpyConfig" in str(config)

    config_str_0 = """
    general:
      n_jobs: 100
    """
    config_100 = AsgardpyConfig.from_yaml(config_str_0)
    assert config_100.general.n_jobs == 100

    with pytest.raises(ValueError):
        config_str_1 = """
        fit_params:
          fit_range:
            min: 10 s
            max: 10 TeV
        """
        AsgardpyConfig.from_yaml(config_str_1)

    with pytest.raises(ValueError):
        config_str_2 = """
        general:
          outdir: ./bla/
        """
        AsgardpyConfig.from_yaml(config_str_2)


def test_config_time():
    """Test for reading Time inputs."""
    from pydantic import ValidationError

    config = AsgardpyConfig()

    with pytest.raises(ValidationError):
        config.dataset1d.instruments[0].dataset_info.observation.obs_time = [
            {"format": "abc", "start": "2000-01-01", "stop": "2001-01-01"}
        ]
    with pytest.raises(ValueError):
        config.dataset1d.instruments[0].dataset_info.observation.obs_time = [
            {"format": "iso", "start": "60000", "stop": "2001-01-01"}
        ]
    with pytest.raises(ValueError):
        config.dataset1d.instruments[0].dataset_info.observation.obs_time = [
            {"format": "iso", "start": "2001-01-01", "stop": "60000"}
        ]


def test_get_model_template():
    """Test for reading a model template by a given tag."""

    from asgardpy.config.operations import get_model_template

    new_model = get_model_template("eclp")

    new_config = AsgardpyConfig.read(new_model)

    assert new_config.target.components[0].spectral.type == "ExpCutoffLogParabolaSpectralModel"

    with pytest.raises(IOError):
        new_config.write(new_model, overwrite=False)


def test_create_config_from_dict():
    """Test to create AsgardpyConfig from a simple dict."""

    gen_dict = {"general": {"log": {"level": "warning"}}}
    config = AsgardpyConfig(**gen_dict)

    assert config.general.log.level == "warning"


@pytest.mark.test_data
def test_config_update_gammapy(gammapy_data_path, base_config_1d):
    """Tests to update target model config from Gammapy-based YAML files."""

    import os

    from asgardpy.config.generator import gammapy_model_to_asgardpy_model_config

    main_config = AsgardpyConfig()

    other_config_path = f"{gammapy_data_path}fermi-3fhl-crab/Fermi-LAT-3FHL_models.yaml"
    other_config_path_2 = f"{gammapy_data_path}estimators/pks2155_hess_lc/models.yaml"

    other_config_1 = gammapy_model_to_asgardpy_model_config(
        other_config_path,
        recursive_merge=False,
    )
    other_config_2 = gammapy_model_to_asgardpy_model_config(
        other_config_path,
        base_config_1d,
        recursive_merge=False,
    )

    main_config.write("test_base_config.yaml", overwrite=True)
    other_config_3 = gammapy_model_to_asgardpy_model_config(
        other_config_path_2,
        "test_base_config.yaml",
        recursive_merge=False,
    )

    main_config_1 = main_config.update(other_config_1)
    main_config_2 = base_config_1d.update(other_config_2)
    main_config_3 = main_config.update(other_config_3)

    new_spectral_model_name = main_config_1.target.components[0].spectral.type
    new_spectral_model_name_2 = main_config_2.target.components[0].spectral.type
    new_spectral_model_name_3 = main_config_3.target.components[0].spectral.type

    index_max = main_config_3.target.components[0].spectral.parameters[0].max

    assert new_spectral_model_name == "LogParabolaSpectralModel"
    assert new_spectral_model_name_2 == "LogParabolaSpectralModel"
    assert new_spectral_model_name_3 == "PowerLawSpectralModel"
    assert index_max == 10.0

    os.remove("test_base_config.yaml")


def test_config_update():
    """Tests to update target model config from other AsgardpyConfig file."""

    from asgardpy.config.operations import get_model_template

    main_config = AsgardpyConfig()

    spec_model_template_file_1 = get_model_template("bpl")
    spec_model_template_file_2 = get_model_template("sbpl")

    other_config_1 = AsgardpyConfig.read(spec_model_template_file_1)
    other_config_2 = AsgardpyConfig.read(spec_model_template_file_2)

    main_config = main_config.update(other_config_1)
    new_spectral_model_name_1 = main_config.target.components[0].spectral.type

    new_config_str = """
    general:
      n_jobs: 100
    """
    main_config_2 = main_config.update(new_config_str)

    main_config = main_config.update(other_config_2, merge_recursive=True)

    new_spectral_model_name_2 = main_config.target.components[0].spectral.type
    spectral_model_params = main_config.target.components[0].spectral.parameters

    assert new_spectral_model_name_1 == "BrokenPowerLawSpectralModel"
    assert main_config_2.general.n_jobs == 100
    with pytest.raises(TypeError):
        main_config.update(5)
    assert new_spectral_model_name_2 == "SmoothBrokenPowerLawSpectralModel"
    assert len(spectral_model_params) == 6


def test_write_model_config():
    """From a Gammapy Models object, write it as an AsgardpyConfig file."""

    from gammapy.modeling.models import (
        ExpCutoffPowerLaw3FGLSpectralModel,
        Models,
        SkyModel,
    )

    from asgardpy.analysis import AsgardpyAnalysis
    from asgardpy.config.generator import AsgardpyConfig, write_asgardpy_model_to_file
    from asgardpy.config.operations import CONFIG_PATH, get_model_template

    config_ = AsgardpyConfig()
    analysis_ = AsgardpyAnalysis(config_)
    model_ = SkyModel(name="Template", spectral_model=ExpCutoffPowerLaw3FGLSpectralModel())

    analysis_.final_model = Models(model_)

    assert get_model_template("ecpl-3fgl")

    write_asgardpy_model_to_file(
        gammapy_model=model_,
        output_file=str(CONFIG_PATH) + "/model_templates/model_template_ecpl-3fgl.yaml",
    )

    with pytest.raises(TypeError):
        write_asgardpy_model_to_file()

    write_asgardpy_model_to_file(
        gammapy_model=model_,
        output_file=None,
    )


1			import pytest
2
3			from asgardpy.config import AsgardpyConfig
4
5
6			def test_config_basic():
7			"""Test on basic Config features."""
8
9			config = AsgardpyConfig()
10			assert "AsgardpyConfig" in str(config)
11
12			config_str_0 = """
13			general:
14			n_jobs: 100
15			"""
16			config_100 = AsgardpyConfig.from_yaml(config_str_0)
17			assert config_100.general.n_jobs == 100
18
19			with pytest.raises(ValueError):
20			config_str_1 = """
21			fit_params:
22			fit_range:
23			min: 10 s
24			max: 10 TeV
25			"""
26			AsgardpyConfig.from_yaml(config_str_1)
27
28			with pytest.raises(ValueError):
29			config_str_2 = """
30			general:
31			outdir: ./bla/
32			"""
33			AsgardpyConfig.from_yaml(config_str_2)
34
35
36			def test_config_time():
37			"""Test for reading Time inputs."""
38			from pydantic import ValidationError
39
40			config = AsgardpyConfig()
41
42			with pytest.raises(ValidationError):
43			config.dataset1d.instruments[0].dataset_info.observation.obs_time = [
44			{"format": "abc", "start": "2000-01-01", "stop": "2001-01-01"}
45			]
46			with pytest.raises(ValueError):
47			config.dataset1d.instruments[0].dataset_info.observation.obs_time = [
48			{"format": "iso", "start": "60000", "stop": "2001-01-01"}
49			]
50			with pytest.raises(ValueError):
51			config.dataset1d.instruments[0].dataset_info.observation.obs_time = [
52			{"format": "iso", "start": "2001-01-01", "stop": "60000"}
53			]
54
55
56			def test_get_model_template():
57			"""Test for reading a model template by a given tag."""
58
59			from asgardpy.config.operations import get_model_template
60
61			new_model = get_model_template("eclp")
62
63			new_config = AsgardpyConfig.read(new_model)
64
65			assert new_config.target.components[0].spectral.type == "ExpCutoffLogParabolaSpectralModel"
66
67			with pytest.raises(IOError):
68			new_config.write(new_model, overwrite=False)
69
70
71			def test_create_config_from_dict():
72			"""Test to create AsgardpyConfig from a simple dict."""
73
74			gen_dict = {"general": {"log": {"level": "warning"}}}
75			config = AsgardpyConfig(**gen_dict)
76
77			assert config.general.log.level == "warning"
78
79
80			@pytest.mark.test_data
81			def test_config_update_gammapy(gammapy_data_path, base_config_1d):
82			"""Tests to update target model config from Gammapy-based YAML files."""
83
84			import os
85
86			from asgardpy.config.generator import gammapy_model_to_asgardpy_model_config
87
88			main_config = AsgardpyConfig()
89
90			other_config_path = f"{gammapy_data_path}fermi-3fhl-crab/Fermi-LAT-3FHL_models.yaml"
91			other_config_path_2 = f"{gammapy_data_path}estimators/pks2155_hess_lc/models.yaml"
92
93			other_config_1 = gammapy_model_to_asgardpy_model_config(
94			other_config_path,
95			recursive_merge=False,
96			)
97			other_config_2 = gammapy_model_to_asgardpy_model_config(
98			other_config_path,
99			base_config_1d,
100			recursive_merge=False,
101			)
102
103			main_config.write("test_base_config.yaml", overwrite=True)
104			other_config_3 = gammapy_model_to_asgardpy_model_config(
105			other_config_path_2,
106			"test_base_config.yaml",
107			recursive_merge=False,
108			)
109
110			main_config_1 = main_config.update(other_config_1)
111			main_config_2 = base_config_1d.update(other_config_2)
112			main_config_3 = main_config.update(other_config_3)
113
114			new_spectral_model_name = main_config_1.target.components[0].spectral.type
115			new_spectral_model_name_2 = main_config_2.target.components[0].spectral.type
116			new_spectral_model_name_3 = main_config_3.target.components[0].spectral.type
117
118			index_max = main_config_3.target.components[0].spectral.parameters[0].max
119
120			assert new_spectral_model_name == "LogParabolaSpectralModel"
121			assert new_spectral_model_name_2 == "LogParabolaSpectralModel"
122			assert new_spectral_model_name_3 == "PowerLawSpectralModel"
123			assert index_max == 10.0
124
125			os.remove("test_base_config.yaml")
126
127
128			def test_config_update():
129			"""Tests to update target model config from other AsgardpyConfig file."""
130
131			from asgardpy.config.operations import get_model_template
132
133			main_config = AsgardpyConfig()
134
135			spec_model_template_file_1 = get_model_template("bpl")
136			spec_model_template_file_2 = get_model_template("sbpl")
137
138			other_config_1 = AsgardpyConfig.read(spec_model_template_file_1)
139			other_config_2 = AsgardpyConfig.read(spec_model_template_file_2)
140
141			main_config = main_config.update(other_config_1)
142			new_spectral_model_name_1 = main_config.target.components[0].spectral.type
143
144			new_config_str = """
145			general:
146			n_jobs: 100
147			"""
148			main_config_2 = main_config.update(new_config_str)
149
150			main_config = main_config.update(other_config_2, merge_recursive=True)
151
152			new_spectral_model_name_2 = main_config.target.components[0].spectral.type
153			spectral_model_params = main_config.target.components[0].spectral.parameters
154
155			assert new_spectral_model_name_1 == "BrokenPowerLawSpectralModel"
156			assert main_config_2.general.n_jobs == 100
157			with pytest.raises(TypeError):
158			main_config.update(5)
159			assert new_spectral_model_name_2 == "SmoothBrokenPowerLawSpectralModel"
160			assert len(spectral_model_params) == 6
161
162
163			def test_write_model_config():
164			"""From a Gammapy Models object, write it as an AsgardpyConfig file."""
165
166			from gammapy.modeling.models import (
167			ExpCutoffPowerLaw3FGLSpectralModel,
168			Models,
169			SkyModel,
170			)
171
172			from asgardpy.analysis import AsgardpyAnalysis
173			from asgardpy.config.generator import AsgardpyConfig, write_asgardpy_model_to_file
174			from asgardpy.config.operations import CONFIG_PATH, get_model_template
175
176			config_ = AsgardpyConfig()
177			analysis_ = AsgardpyAnalysis(config_)
178			model_ = SkyModel(name="Template", spectral_model=ExpCutoffPowerLaw3FGLSpectralModel())
179
180			analysis_.final_model = Models(model_)
181
182			assert get_model_template("ecpl-3fgl")
183
184			write_asgardpy_model_to_file(
185			gammapy_model=model_,
186			output_file=str(CONFIG_PATH) + "/model_templates/model_template_ecpl-3fgl.yaml",
187			)
188
189			with pytest.raises(TypeError):
190			write_asgardpy_model_to_file()
191
192			write_asgardpy_model_to_file(
193			gammapy_model=model_,
194			output_file=None,
195			)
196

chaimain / asgardpy

Push — main ( 4744cf...d735dd )

test_write_model_config() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like