Commit e16f8a88 authored by Tapani Pihlajakuja's avatar Tapani Pihlajakuja
Browse files

Add support for update_ref in MASA tests.

parent 5c4823b9

tests/test_masa_enc_dec.py

+82 −76
Original line number Diff line number Diff line
@@ -71,6 +71,7 @@ def check_and_makedir(dir_path): 
                raise  # raises the error again





@pytest.mark.create_ref

@pytest.mark.parametrize("output_mode", output_mode_list)

@pytest.mark.parametrize("ivas_br", ivas_br_masa)

@pytest.mark.parametrize("masa_metadata_audio_ndir_ntransportch_dtx", masa_metadata_audio_ndir_ntransportch_dtx_list)
@@ -81,6 +82,7 @@ def test_masa_enc_dec( 
        ref_decoder_path,

        reference_path,

        dut_base_path,

        update_ref,

        keep_files,

        ivas_br,

        masa_metadata_audio_ndir_ntransportch_dtx,
@@ -131,6 +133,7 @@ def test_masa_enc_dec( 
        dec_met_output_ref = None

        dec_met_output_dut = None



    if update_ref == 2 or update_ref == 1:

        # Encode REF

        ivas_enc(

            ref_encoder_frontend,
@@ -143,6 +146,16 @@ def test_masa_enc_dec( 
            dtx,

        )



        # Decode REF

        ivas_dec(

            ref_decoder_frontend,

            output_mode,

            out_fs,

            output_bitstream_ref,

            dec_output_ref,

        )



    if update_ref == 2 or update_ref == 0:

        # Encode DUT

        ivas_enc(

            dut_encoder_frontend,
@@ -155,15 +168,6 @@ def test_masa_enc_dec( 
            dtx,

        )



    # Decode REF

    ivas_dec(

        ref_decoder_frontend,

        output_mode,

        out_fs,

        output_bitstream_ref,

        dec_output_ref,

    )



        # Decode DUT

        ivas_dec(

            dut_decoder_frontend,
@@ -188,6 +192,8 @@ def test_masa_enc_dec( 
                pytest.fail("Metadata output difference detected")

            elif pcmcmp_res != 0:

                pytest.fail("Transport output difference detected")

            else:

                print("Comparison bit exact")            



        else:

            # Compare audio outputs
@@ -196,18 +202,18 @@ def test_masa_enc_dec( 
                cmp_result, reason = cmp_pcm(dec_output_dut, dec_output_ref, output_mode, int(out_fs*1000))

                # Report compare result

                assert cmp_result == 0, reason







    remove_output(

        keep_files,

        output_bitstream_ref,

        output_bitstream_dut,

        dec_output_ref,

        dec_output_dut,

        dec_met_output_ref,

        dec_met_output_dut,

    )

            else:

                print("Comparison bit exact")            



        # remove_output(

        #     keep_files,

        #     output_bitstream_ref,

        #     output_bitstream_dut,

        #     dec_output_ref,

        #     dec_output_dut,

        #     dec_met_output_ref,

        #     dec_met_output_dut,

        # )





#########################################################