mirror of
https://github.com/Telecominfraproject/oopt-gnpy.git
synced 2026-01-27 02:21:40 +00:00
ff7ce224f7
- removing order dependancy in the init - adding test dedicated to the add method - fix invocation test multiband Change-Id: Ibaf8b6d791c17136b2c3538a5ce82e0e2757b25b
145 lines
6.4 KiB
Python
145 lines
6.4 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
# test_info
|
|
# Copyright (C) 2025 Telecom Infra Project and GNPy contributors
|
|
# see AUTHORS.rst for a list of contributors
|
|
|
|
"""
|
|
Checks spectral information utilities
|
|
"""
|
|
|
|
import pytest
|
|
from numpy import array, zeros, ones, linspace
|
|
from numpy.testing import assert_array_equal
|
|
from gnpy.core.info import create_arbitrary_spectral_information, create_input_spectral_information
|
|
from gnpy.core.exceptions import SpectrumError
|
|
from gnpy.core.utils import dbm2watt
|
|
|
|
|
|
def test_create_arbitrary_spectral_information():
|
|
si = create_arbitrary_spectral_information(frequency=[193.25e12, 193.3e12, 193.35e12],
|
|
baud_rate=32e9, pch=[1, 1, 1],
|
|
delta_pdb_per_channel=[1, 1, 1],
|
|
tx_osnr=40.0, tx_power=[1, 1, 1])
|
|
assert_array_equal(si.baud_rate, array([32e9, 32e9, 32e9]))
|
|
assert_array_equal(si.slot_width, array([37.5e9, 37.5e9, 37.5e9]))
|
|
assert_array_equal(si.signal, ones(3))
|
|
assert_array_equal(si.nli, zeros(3))
|
|
assert_array_equal(si.ase, zeros(3))
|
|
assert_array_equal(si.delta_pdb_per_channel, ones(3))
|
|
assert_array_equal(si.roll_off, zeros(3))
|
|
assert_array_equal(si.chromatic_dispersion, zeros(3))
|
|
assert_array_equal(si.pmd, zeros(3))
|
|
assert_array_equal(si.channel_number, array([1, 2, 3]))
|
|
assert_array_equal(si.number_of_channels, 3)
|
|
assert_array_equal(si.df, array([[0, 50e9, 100e9], [-50e9, 0, 50e9], [-100e9, -50e9, 0]]))
|
|
assert_array_equal(si.tx_osnr, array([40.0, 40.0, 40.0]))
|
|
|
|
with pytest.raises(SpectrumError, match='Spectra cannot be summed: channels overlapping.'):
|
|
si += si
|
|
|
|
si = create_arbitrary_spectral_information(frequency=array([193.35e12, 193.3e12, 193.25e12]),
|
|
slot_width=array([50e9, 50e9, 50e9]),
|
|
baud_rate=32e9, pch=array([1, 2, 3]),
|
|
tx_osnr=40.0, tx_power=array([1, 2, 3]))
|
|
|
|
assert_array_equal(si.signal, array([3, 2, 1]))
|
|
|
|
with pytest.raises(SpectrumError, match='Spectrum baud rate, including the roll off, '
|
|
r'larger than the slot width for channels: \[1, 3\].'):
|
|
create_arbitrary_spectral_information(frequency=[193.25e12, 193.3e12, 193.35e12], pch=1,
|
|
baud_rate=[64e9, 32e9, 64e9], slot_width=50e9,
|
|
tx_osnr=40.0, tx_power=1)
|
|
with pytest.raises(SpectrumError, match='Spectrum required slot widths larger than the frequency spectral '
|
|
r'distances between channels: \[\(1, 2\), \(3, 4\)\].'):
|
|
create_arbitrary_spectral_information(frequency=[193.26e12, 193.3e12, 193.35e12, 193.39e12], pch=1,
|
|
tx_osnr=40.0, baud_rate=32e9, slot_width=50e9, tx_power=1)
|
|
with pytest.raises(SpectrumError, match='Spectrum required slot widths larger than the frequency spectral '
|
|
r'distances between channels: \[\(1, 2\), \(2, 3\)\].'):
|
|
create_arbitrary_spectral_information(frequency=[193.25e12, 193.3e12, 193.35e12], pch=1, baud_rate=49e9,
|
|
tx_osnr=40.0, roll_off=0.1, tx_power=1)
|
|
with pytest.raises(SpectrumError,
|
|
match='Dimension mismatch in input fields.'):
|
|
create_arbitrary_spectral_information(frequency=[193.25e12, 193.3e12, 193.35e12], pch=[1, 2], baud_rate=49e9,
|
|
tx_osnr=40.0, tx_power=1)
|
|
|
|
|
|
def test_noise_ratios():
|
|
tx_power = 0.001
|
|
si = create_input_spectral_information(f_min=191.5e12, f_max=196e12, baud_rate=32e9, spacing=50e9, roll_off=0.1,
|
|
tx_osnr=40.0, tx_power=tx_power)
|
|
|
|
assert all(si.pch == tx_power)
|
|
assert all(si.signal == si.pch)
|
|
assert all(si._signal_ratio == 1)
|
|
assert all(si._ase_ratio == 0)
|
|
assert all(si._nli_ratio == 0)
|
|
|
|
add_ase = tx_power / 100
|
|
pch_in = si.pch
|
|
sig_in = si.signal
|
|
ase_in = si.ase
|
|
nli_in = si.nli
|
|
|
|
si.add_ase(add_ase)
|
|
|
|
assert all(si.pch == pch_in + add_ase)
|
|
assert all(si.signal == sig_in)
|
|
assert all(si.ase == ase_in + add_ase)
|
|
assert all(si.nli == nli_in)
|
|
assert all(si._signal_ratio + si._ase_ratio + si._nli_ratio == 1)
|
|
|
|
add_nli = tx_power / 200
|
|
pch_in = si.pch
|
|
sig_in = si.signal
|
|
ase_in = si.ase
|
|
nli_in = si.nli
|
|
|
|
si.add_nli(add_nli)
|
|
|
|
assert all(si.pch == pch_in)
|
|
assert all(si.signal == sig_in * (1 - add_nli / pch_in))
|
|
assert all(si.ase == ase_in * (1 - add_nli / pch_in))
|
|
assert all(si.nli == nli_in * (1 - add_nli / pch_in) + add_nli)
|
|
assert all(si._signal_ratio + si._ase_ratio + si._nli_ratio == 1)
|
|
|
|
|
|
def test_spectral_information_add():
|
|
"""
|
|
Testing Spectral Information add method.
|
|
"""
|
|
n_ch = 40
|
|
grid = 100e9
|
|
baud_rate = 32e9
|
|
delta_pdb_per_channel = 0.0
|
|
tx_osnr = 40.0
|
|
|
|
# C band
|
|
start_frequency_c_band = 191.5e12
|
|
frequency_c_band = array([start_frequency_c_band + i * grid for i in range(n_ch)])
|
|
pch_c = dbm2watt(linspace(-1.0, 1.0, n_ch))
|
|
|
|
si_c_band = create_arbitrary_spectral_information(frequency=frequency_c_band,
|
|
baud_rate=baud_rate, pch=pch_c,
|
|
delta_pdb_per_channel=delta_pdb_per_channel, tx_osnr=tx_osnr)
|
|
|
|
# L band
|
|
start_frequency_l_band = 185.5e12
|
|
frequency_l_band = array([start_frequency_l_band + i * grid for i in range(n_ch)])
|
|
pch_l = dbm2watt(linspace(1.0, -1.0, n_ch))
|
|
|
|
si_l_band = create_arbitrary_spectral_information(frequency=frequency_l_band,
|
|
baud_rate=baud_rate, pch=pch_l,
|
|
delta_pdb_per_channel=delta_pdb_per_channel, tx_osnr=tx_osnr)
|
|
|
|
# C+L spectrum
|
|
si_c_plus_l = si_c_band + si_l_band
|
|
si_l_plus_c = si_l_band + si_c_band
|
|
|
|
assert_array_equal(si_c_plus_l.pch, si_l_plus_c.pch)
|
|
assert_array_equal(si_c_plus_l.channel_number, si_l_plus_c.channel_number)
|
|
assert_array_equal(si_c_plus_l.number_of_channels, si_l_plus_c.number_of_channels)
|
|
assert_array_equal(si_c_plus_l.df, si_l_plus_c.df)
|