mirror of
https://github.com/Telecominfraproject/oopt-gnpy.git
synced 2025-10-30 17:47:50 +00:00
7a1b15a916
Signed-off-by: EstherLerouzic <esther.lerouzic@orange.com> Change-Id: Ifdd6a566fda74c5b7d417f9d61c51d4d3da07bfd
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
# test_parameters
|
|
# Copyright (C) 2025 Telecom Infra Project and GNPy contributors
|
|
# see AUTHORS.rst for a list of contributors
|
|
|
|
|
|
"""
|
|
Checks that the class SimParams behaves as a mutable Singleton.
|
|
"""
|
|
|
|
import pytest
|
|
from pathlib import Path
|
|
from numpy.testing import assert_allclose
|
|
|
|
from gnpy.core.parameters import SimParams, FiberParams
|
|
from gnpy.tools.json_io import load_json, Fiber
|
|
|
|
TEST_DIR = Path(__file__).parent
|
|
|
|
|
|
@pytest.mark.usefixtures('set_sim_params')
|
|
def test_sim_parameters():
|
|
sim_params = {'nli_params': {}, 'raman_params': {}}
|
|
SimParams.set_params(sim_params)
|
|
s1 = SimParams()
|
|
assert s1.nli_params.method == 'gn_model_analytic'
|
|
s2 = SimParams()
|
|
assert not s1.raman_params.flag
|
|
sim_params['raman_params']['flag'] = True
|
|
SimParams.set_params(sim_params)
|
|
assert s2.raman_params.flag
|
|
assert s1.raman_params.flag
|
|
|
|
|
|
def test_fiber_parameters():
|
|
fiber_dict_explicit_g0 = load_json(TEST_DIR/'data'/'test_parameters_fiber_config.json')['params']
|
|
fiber_params_explicit_g0 = FiberParams(**fiber_dict_explicit_g0)
|
|
|
|
fiber_dict_default_g0 = load_json(TEST_DIR/'data'/'test_science_utils_fiber_config.json')['params']
|
|
fiber_params_default_g0 = FiberParams(**fiber_dict_default_g0)
|
|
|
|
fiber_dict_cr = load_json(TEST_DIR/'data'/'test_old_parameters_fiber_config.json')['params']
|
|
fiber_dict_cr.update(Fiber(**fiber_dict_cr).__dict__)
|
|
fiber_params_cr = FiberParams(**fiber_dict_cr)
|
|
|
|
norm_gamma_raman_explicit_g0 = fiber_params_explicit_g0.raman_coefficient.normalized_gamma_raman
|
|
norm_gamma_raman_default_g0 = fiber_params_default_g0.raman_coefficient.normalized_gamma_raman
|
|
|
|
norm_gamma_raman_cr = fiber_params_cr.raman_coefficient.normalized_gamma_raman
|
|
|
|
assert_allclose(norm_gamma_raman_explicit_g0, norm_gamma_raman_default_g0, rtol=1e-10)
|
|
assert_allclose(norm_gamma_raman_explicit_g0, norm_gamma_raman_cr, rtol=1e-10)
|
|
|
|
# Change Effective Area
|
|
fiber_dict_default_g0['effective_area'] = 100e-12
|
|
no_ssmf_fiber_params = FiberParams(**fiber_dict_default_g0)
|
|
|
|
norm_gamma_raman_default_g0_no_ssmf = no_ssmf_fiber_params.raman_coefficient.normalized_gamma_raman
|
|
|
|
assert_allclose(norm_gamma_raman_explicit_g0, norm_gamma_raman_default_g0_no_ssmf, rtol=1e-10)
|