Commit de2c3937 authored by Archit Tamarapu's avatar Archit Tamarapu
Browse files

add external split rendering tests for OMASA and OSBA

parent c72f64d6

tests/split_rendering/test_split_rendering.py

+98 −28
Original line number Diff line number Diff line
@@ -383,6 +383,41 @@ def test_omasa_full_chain_split( 
    )





@pytest.mark.parametrize("trajectory", SPLIT_REND_HR_TRAJECTORIES_TO_TEST)

@pytest.mark.parametrize("render_config", RENDERER_CONFIGS_TO_TEST_OMASA)

@pytest.mark.parametrize("in_fmt", INPUT_FORMATS_OMASA_SPLIT_REND)

def test_omasa_external_split(

    record_property,

    props_to_record,

    get_mld,

    get_mld_lim,

    get_ssnr,

    get_odg,

    get_odg_bin,

    test_info,

    in_fmt,

    render_config,

    trajectory,

):

    post_trajectory = HR_TRAJECTORY_DIR.joinpath(f"{trajectory}.csv")

    pre_trajectory = post_trajectory.with_stem(f"{post_trajectory.stem}_delayed")



    run_external_split_rendering(

        record_property,

        props_to_record,

        test_info,

        in_fmt=in_fmt,

        render_config=RENDER_CFG_DIR.joinpath(f"{render_config}.txt"),

        pre_trajectory=pre_trajectory,

        post_trajectory=post_trajectory,

        get_mld=get_mld,

        mld_lim=get_mld_lim,

        get_ssnr=get_ssnr,

        get_odg=get_odg,

        get_odg_bin=get_odg_bin,

    )





""" OSBA """
@@ -425,6 +460,41 @@ def test_osba_full_chain_split( 
    )





@pytest.mark.parametrize("trajectory", SPLIT_REND_HR_TRAJECTORIES_TO_TEST)

@pytest.mark.parametrize("render_config", RENDERER_CONFIGS_TO_TEST_OSBA)

@pytest.mark.parametrize("in_fmt", INPUT_FORMATS_OSBA_SPLIT_REND)

def test_osba_external_split(

    record_property,

    props_to_record,

    get_mld,

    get_mld_lim,

    get_ssnr,

    get_odg,

    get_odg_bin,

    test_info,

    in_fmt,

    render_config,

    trajectory,

):

    post_trajectory = HR_TRAJECTORY_DIR.joinpath(f"{trajectory}.csv")

    pre_trajectory = post_trajectory.with_stem(f"{post_trajectory.stem}_delayed")



    run_external_split_rendering(

        record_property,

        props_to_record,

        test_info,

        in_fmt=in_fmt,

        render_config=RENDER_CFG_DIR.joinpath(f"{render_config}.txt"),

        pre_trajectory=pre_trajectory,

        post_trajectory=post_trajectory,

        get_mld=get_mld,

        mld_lim=get_mld_lim,

        get_ssnr=get_ssnr,

        get_odg=get_odg,

        get_odg_bin=get_odg_bin,

    )





""" PLC """
@@ -474,7 +544,7 @@ full_chain_split_pcm_params = [ 
    ("7_1_4", "512000", "split_renderer_config_3dofhq_512k_lc3plus"),

    ("ISM4", "384000", "split_renderer_config_2dof_768k_default"),

    ("MASA2", "256000", "split_renderer_config_3dof_384k_lcld"),

    ("ISM2MASA4", "256000", "split_renderer_config_3dof_384k_lcld"),

    ("ISM4MASA2", "256000", "split_renderer_config_3dof_384k_lcld"),

]

tests/split_rendering/utils.py

+7 −2
Original line number Diff line number Diff line
@@ -225,7 +225,6 @@ def run_full_chain_split_rendering( 
        elif (

            in_fmt.upper().startswith("ISM")

            or in_fmt.upper().startswith("MASA")

            or in_fmt.upper().startswith("OMASA")

        ):

            in_meta_files = FORMAT_TO_METADATA_FILES[in_fmt]

        else:
@@ -386,7 +385,13 @@ def run_external_split_rendering( 
            )



        # check for metadata files

        if in_fmt.upper().startswith("ISM") or in_fmt.upper().startswith("MASA"):

        if in_fmt.upper().startswith("ISM") and "SBA" in in_fmt.upper():

            # use name MD as ISM

            in_meta_files = FORMAT_TO_METADATA_FILES[f"ISM{in_fmt[3]}"]

        elif (

            in_fmt.upper().startswith("ISM")

            or in_fmt.upper().startswith("MASA")

        ):

            in_meta_files = FORMAT_TO_METADATA_FILES[in_fmt]

        else:

            in_meta_files = None