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

re-use keep_files flag in selection-BE test

parent eaeb6331

tests/codec_be_on_mr_selection/__init__.py

+6 −0
Original line number Diff line number Diff line
@@ -126,6 +126,7 @@ def run_check( 
    decoder_frontend,

    is_ref_creation,

    input_file_num=None,

    keep_files=True,

):

    sampling_rate = 48

    output_mode, options = OUTPUT_MODES_AND_OPTIONS_FOR_EXPERIMENT[experiment]
@@ -182,8 +183,13 @@ def run_check( 
    if not is_ref_creation:

        if not is_be_to_ref(dut_output):

            pytest.fail("Decoder output differs from reference")

        elif not keep_files:

            os.remove(dut_output)

            os.remove(dut_bitstream)

        for md in metadata:

            md_suffix = "".join(md.suffixes)

            dut_md = DUT_PATH.joinpath(dut_output.with_suffix(md_suffix).name)

            if not is_be_to_ref(dut_md):

                pytest.fail("Metadata file {md.name} differs from reference")

            elif not keep_files:

                os.remove(dut_md)

tests/codec_be_on_mr_selection/test_experiments.py

+6 −0
Original line number Diff line number Diff line
@@ -52,6 +52,7 @@ def test_p800( 
    dut_encoder_frontend,

    dut_decoder_frontend,

    update_ref,

    keep_files,

):

    run_check(

        experiment,
@@ -63,6 +64,7 @@ def test_p800( 
        dut_encoder_frontend,

        dut_decoder_frontend,

        update_ref == 1,

        keep_files=keep_files,

    )
@@ -79,6 +81,7 @@ def test_bs1534_no_masa( 
    dut_encoder_frontend,

    dut_decoder_frontend,

    update_ref,

    keep_files,

):

    category = ""

    run_check(
@@ -92,6 +95,7 @@ def test_bs1534_no_masa( 
        dut_decoder_frontend,

        update_ref == 1,

        input_file_num=input_file_num,

        keep_files=keep_files,

    )
@@ -111,6 +115,7 @@ def test_bs1534_masa( 
    dut_encoder_frontend,

    dut_decoder_frontend,

    update_ref,

    keep_files,

):

    run_check(

        experiment,
@@ -123,4 +128,5 @@ def test_bs1534_masa( 
        dut_decoder_frontend,

        update_ref == 1,

        input_file_num=input_file_num,

        keep_files=keep_files,

    )