Commit 81f977c7 authored by Jan Kiene's avatar Jan Kiene
Browse files

implement --decoder_only for masa tests

parent 66f4cdf7
Loading
Loading
Loading
Loading
+16 −22
Original line number Diff line number Diff line
@@ -109,6 +109,7 @@ def test_masa_enc_dec(
    output_mode,
    get_mld,
    get_mld_lim,
    decoder_only,
):
    # Input parameters
    in_fs = 48
@@ -153,7 +154,7 @@ def test_masa_enc_dec(
        dec_met_output_ref = None
        dec_met_output_dut = None

    if update_ref == 2 or update_ref == 1:
    if (update_ref == 2 or update_ref == 1):
        # Encode REF
        ivas_enc(
            ref_encoder_frontend,
@@ -176,6 +177,9 @@ def test_masa_enc_dec(
        )

    if update_ref == 2 or update_ref == 0:
        if decoder_only:
            output_bitstream_dut = output_bitstream_ref
        else:
            # Encode DUT
            ivas_enc(
                dut_encoder_frontend,
@@ -256,16 +260,6 @@ def test_masa_enc_dec(
                    record_property("MLD", "0")
                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,
        # )


#########################################################
# -------------------- test function --------------------