Commit 57d3461b authored by Archit Tamarapu's avatar Archit Tamarapu
Browse files

[fix] formatting + use shorter testvectors for smoketest

parent ef9f5155

tests/split_rendering/constants.py

+1 −0
Original line number Diff line number Diff line
@@ -47,6 +47,7 @@ sys.path.append(TESTS_DIR) 
from renderer.constants import (

    CUSTOM_LS_TO_TEST,

    FORMAT_TO_FILE_COMPARETEST,

    FORMAT_TO_FILE_SMOKETEST,

    FORMAT_TO_METADATA_FILES,

    HR_TRAJECTORIES_TO_TEST,

    INPUT_FORMATS_AMBI,

tests/split_rendering/utils.py

+14 −2
Original line number Diff line number Diff line
@@ -51,6 +51,7 @@ def run_full_chain_split_rendering( 
    post_trajectory: Path,

    output_path_base: Path,

    binary_suffix: str = "",

    is_comparetest: bool = False,

) -> Path:

    """

    Runs the full split rendering chain consisting of
@@ -74,7 +75,10 @@ def run_full_chain_split_rendering( 
        cmd = SPLIT_PRE_COD_CMD[:]



        cmd[0] += binary_suffix

        if is_comparetest:

            cmd[3] = str(FORMAT_TO_FILE_COMPARETEST[in_fmt])

        else:

            cmd[3] = str(FORMAT_TO_FILE_SMOKETEST[in_fmt])

        cmd[4] = str(ivas_bitstream)

        cmd[1:1] = FORMAT_TO_IVAS_COD_FORMAT[in_fmt]
@@ -114,6 +118,7 @@ def run_external_split_rendering( 
    post_trajectory: Path,

    output_path_base: Path,

    binary_suffix: str = "",

    is_comparetest: bool = False,

) -> Path:

    """

    Runs the exeternal split rendering chain consisting of
@@ -137,7 +142,10 @@ def run_external_split_rendering( 
        cmd = SPLIT_PRE_REND_CMD[:]

        cmd[0] += binary_suffix

        cmd[4] = str(render_config)

        if is_comparetest:

            cmd[6] = str(FORMAT_TO_FILE_COMPARETEST[in_fmt])

        else:

            cmd[6] = str(FORMAT_TO_FILE_SMOKETEST[in_fmt])

        cmd[8] = in_fmt

        cmd[10] = str(split_bitstream)

        cmd[14] = str(pre_trajectory)
@@ -167,10 +175,12 @@ def compare_full_chain_split_args( 
    ref, ref_fs = run_full_chain_split_rendering(

        output_path_base=OUTPUT_PATH_REF,

        binary_suffix=BIN_SUFFIX_MERGETARGET,

        is_comparetest=True,

        **kwargs,

    )

    cut, cut_fs = run_full_chain_split_rendering(

        output_path_base=OUTPUT_PATH_CUT,

        is_comparetest=True,

        **kwargs,

    )

    check_BE(test_info, ref, ref_fs, cut, cut_fs)
@@ -180,10 +190,12 @@ def compare_external_split_args(test_info, **kwargs): 
    ref, ref_fs = run_external_split_rendering(

        output_path_base=OUTPUT_PATH_REF,

        binary_suffix=BIN_SUFFIX_MERGETARGET,

        is_comparetest=True,

        **kwargs,

    )

    cut, cut_fs = run_external_split_rendering(

        output_path_base=OUTPUT_PATH_CUT,

        is_comparetest=True,

        **kwargs,

    )

    check_BE(test_info, ref, ref_fs, cut, cut_fs)

tests/split_rendering/test_split_rendering.py

+1 −1

File changed.

Contains only whitespace changes.

tests/split_rendering/test_split_rendering_be_comparison.py

+1 −1

File changed.

Contains only whitespace changes.