Commit 81436f19 authored by Jiaquan Huo's avatar Jiaquan Huo
Browse files

add ltv to sba pytests, not tested

parent 4a4bb894

tests/codec_be_on_mr_nonselection/test_sba.py

+45 −10
Original line number Diff line number Diff line
@@ -46,11 +46,11 @@ from ..constants import TESTV_DIR 


# params



tag_list = ["stvFOA"]

tag_list_HOA2 = ["stv2OA"]

tag_list_HOA3 = ["stv3OA"]

tag_list = ["stvFOA", "ltvFOA"]

tag_list_HOA2 = ["stv2OA", "ltvHOA2"]

tag_list_HOA3 = ["stv3OA", "ltvHOA3"]



tag_list_bw_force = ["stvFOA"]

tag_list_bw_force = ["stvFOA", "ltvFOA"]

dtx_set = ["0", "1"]

dict_fsample_bw = {"48": "3", "32": "2", "16": "1"}

dict_bw_idx = {"FB": "3", "SWB": "2", "WB": "1"}
@@ -117,7 +117,6 @@ def test_pca_enc( 
    get_odg,

):

    pca = True

    tag = tag + fs + "c"

    ivas_br = "256000"

    dtx = "0"

    max_bw = "FB"
@@ -125,6 +124,13 @@ def test_pca_enc( 
    sba_order = "+1"

    output_config = "FOA"



    if "ltv" in tag:

        tag = f"ltv{fs}_FOA"

    elif "stv" in tag:

        tag = tag + fs + "c"

    else:

        assert 0 



    if not decoder_only:

        # enc

        sba_enc(
@@ -227,7 +233,14 @@ def test_sba_enc_system( 
        pytest.skip()

    if gain_flag == 1 and ivas_br not in ["13200", "16400", "24400", "32000"]:

        pytest.skip()



    if "ltv" in tag:

        tag = f"ltv{fs}_FOA"

    elif "stv" in tag:

        tag = tag + fs + "c"

    else:

        assert 0 



    max_bw = "FB"

    sba_order = "+1"

    output_config = "FOA"
@@ -317,7 +330,13 @@ def test_spar_hoa2_enc_system( 
    dtx = "0"

    gain_flag = -1



    if "ltv" in tag:

        tag = f"ltv{fs}_HOA2"

    elif "stv" in tag:

        tag = tag + fs + "c"

    else:

        assert 0 



    max_bw = "FB"

    sba_order = "+2"

    output_config = "HOA2"
@@ -398,7 +417,13 @@ def test_spar_hoa3_enc_system( 
    dtx = "0"

    gain_flag = -1



    if "ltv" in tag:

        tag = f"ltv{fs}_HOA3"

    elif "stv" in tag:

        tag = tag + fs + "c"

    else:

        assert 0 



    max_bw = "FB"

    sba_order = "+3"

    output_config = "HOA3"
@@ -488,7 +513,12 @@ def test_sba_enc_BWforce_system( 
        pytest.skip()

    fs = sample_rate_bw_idx[0]

    bw = sample_rate_bw_idx[1]

    if "ltv" in tag:

        tag = f"ltv{fs}_FOA"

    elif "stv" in tag:

        tag = tag + fs + "c"

    else:

        assert 0 

    gain_flag = -1

    sba_order = "+1"

    output_config = "FOA"
@@ -593,7 +623,12 @@ def test_sba_plc_system( 
            pytest.skip()

    if gain_flag == 1 and ivas_br not in ["13200", "16400", "24400", "32000"]:

        pytest.skip()

    if "ltv" in tag:

        tag = f"ltv{fs}_FOA"

    elif "stv" in tag:

        tag = tag + fs + "c"

    else:

        assert 0 



    # dec

    sba_dec(