Commit fccf1155 authored by Jan Kiene's avatar Jan Kiene
Browse files

record SSNR as property

parent fdf2af96

tests/codec_be_on_mr_nonselection/test_masa_enc_dec.py

+8 −1
Original line number Diff line number Diff line
@@ -44,7 +44,7 @@ import pytest 
from tests.cmp_pcm import cmp_pcm

from tests.conftest import DecoderFrontend, EncoderFrontend



from ..constants import MLD_PATTERN, MAX_DIFF_PATTERN

from ..constants import MLD_PATTERN, MAX_DIFF_PATTERN, SSNR_PATTERN



# params

# output_mode_list = ['MONO', 'STEREO', '5_1', '7_1', '5_1_2', '5_1_4', '7_1_4', 'FOA', 'HOA2', 'HOA3', 'BINAURAL', 'BINAURAL_ROOM', 'EXT']
@@ -226,6 +226,13 @@ def test_masa_enc_dec( 
                mld = re.search(MLD_PATTERN, reason).groups(1)[0]

                record_property("MLD", mld)



            if get_ssnr:

                ssnrs = re.findall(SSNR_PATTERN, reason)

                min_ssnr = min(ssnrs)

                min_ssnr_channel = ssnrs.index(min_ssnr)

                record_property("MIN_SSNR", min_ssnr)

                record_property("MIN_SSNR_CHANNEL", min_ssnr_channel)



            max_diff = 0

            if pcmcmp_res:

                search_result = re.search(MAX_DIFF_PATTERN, reason)

tests/codec_be_on_mr_nonselection/test_param_file.py

+8 −1
Original line number Diff line number Diff line
@@ -45,7 +45,7 @@ import numpy as np 
from tests.cmp_pcm import cmp_pcm

from tests.conftest import DecoderFrontend, EncoderFrontend

from tests.testconfig import PARAM_FILE

from ..constants import MLD_PATTERN, MAX_DIFF_PATTERN

from ..constants import MLD_PATTERN, MAX_DIFF_PATTERN, SSNR_PATTERN



VALID_DEC_OUTPUT_CONF = [

    "MONO",
@@ -358,6 +358,13 @@ def test_param_file_tests( 
            mld = re.search(MLD_PATTERN, reason).groups(1)[0]

            record_property("MLD", mld)



        if get_ssnr:

            ssnrs = re.findall(SSNR_PATTERN, reason)

            min_ssnr = min(ssnrs)

            min_ssnr_channel = ssnrs.index(min_ssnr)

            record_property("MIN_SSNR", min_ssnr)

            record_property("MIN_SSNR_CHANNEL", min_ssnr_channel)



        max_diff = 0

        if output_differs:

            search_result = re.search(MAX_DIFF_PATTERN, reason)

tests/codec_be_on_mr_nonselection/test_sba_bs_dec_plc.py

+8 −1
Original line number Diff line number Diff line
@@ -40,7 +40,7 @@ import pytest 


from tests.cmp_pcm import cmp_pcm

from tests.conftest import DecoderFrontend

from ..constants import MLD_PATTERN, MAX_DIFF_PATTERN

from ..constants import MLD_PATTERN, MAX_DIFF_PATTERN, SSNR_PATTERN



# params

tag_list = ["stvFOA"]
@@ -222,6 +222,13 @@ def sba_dec_plc( 
            mld = re.search(MLD_PATTERN, reason).groups(1)[0]

            record_property("MLD", mld)



        if get_ssnr:

            ssnrs = re.findall(SSNR_PATTERN, reason)

            min_ssnr = min(ssnrs)

            min_ssnr_channel = ssnrs.index(min_ssnr)

            record_property("MIN_SSNR", min_ssnr)

            record_property("MIN_SSNR_CHANNEL", min_ssnr_channel)



        max_diff = 0

        if cmp_result:

            search_result = re.search(MAX_DIFF_PATTERN, reason)

tests/codec_be_on_mr_nonselection/test_sba_bs_enc.py

+8 −1
Original line number Diff line number Diff line
@@ -42,7 +42,7 @@ from cut_bs import cut_from_start 


from tests.cmp_pcm import cmp_pcm

from tests.conftest import DecoderFrontend, EncoderFrontend

from ..constants import MLD_PATTERN, MAX_DIFF_PATTERN

from ..constants import MLD_PATTERN, MAX_DIFF_PATTERN, SSNR_PATTERN



# params
@@ -746,6 +746,13 @@ def sba_dec( 
            mld = re.search(MLD_PATTERN, reason).groups(1)[0]

            record_property("MLD", mld)



        if get_ssnr:

            ssnrs = re.findall(SSNR_PATTERN, reason)

            min_ssnr = min(ssnrs)

            min_ssnr_channel = ssnrs.index(min_ssnr)

            record_property("MIN_SSNR", min_ssnr)

            record_property("MIN_SSNR_CHANNEL", min_ssnr_channel)



        max_diff = 0

        if cmp_result:

            search_result = re.search(MAX_DIFF_PATTERN, reason)

tests/constants.py

+1 −0
Original line number Diff line number Diff line 
# regex patterns for parsing the output from cmp_pcm -> mainly for BASOP ci

MLD_PATTERN = r"MLD: ([\d\.]*)"

MAX_DIFF_PATTERN = r"MAXIMUM ABS DIFF: (\d*)"

SSNR_PATTERN = r"Channel \d* SSNR: (\d*\.\d*)"