Commit 1468ce62 authored by Marek Szczerba's avatar Marek Szczerba
Browse files

Updated RTP tests for acoustic environment (prototype)

parent e1736a19

tests/rtp/test_rtp.py

+24 −3
Original line number Original line Diff line number Diff line
@@ -228,6 +228,23 @@ def generatePiData(startTs: int, endTs: int) -> dict: 
    someAuFocusLvl = lambda: AUDIO_FOCUS(level=AUDIO_FOCUS_LEVEL(random.randint(0, 15)))

    someAuFocusLvl = lambda: AUDIO_FOCUS(level=AUDIO_FOCUS_LEVEL(random.randint(0, 15)))

    someAuFocusList = [someAuFocusDirLvl, someAuFocusDir, someAuFocusLvl]

    someAuFocusList = [someAuFocusDirLvl, someAuFocusDir, someAuFocusLvl]





    someAcousticEnvAEID = lambda: ACOUSTIC_ENVIRONMENT(

        aeid=random.randint(0, 127)

    )

    someAcousticEnvLR = lambda: ACOUSTIC_ENVIRONMENT(

        aeid=random.randint(0, 127),

        rt60=[random.randint(0, len(rt60Value)) for _ in range(3)],

        dsr=[random.randint(0, len(dsrValue)) for _ in range(3)],

    )

    someAcousticEnvERLR = lambda: ACOUSTIC_ENVIRONMENT(

        aeid=random.randint(0, 127),

        rt60=[random.randint(0, len(rt60Value)) for _ in range(3)],

        dsr=[random.randint(0, len(dsrValue)) for _ in range(3)],

        dim=[random.randint(0, len(roomDimensionValue)) for _ in range(3)],

        abscoeff=[random.randint(0, len(absorptionCoeffValues)) for _ in range(6)],

    )

    someAcousticEnvList = [someAcousticEnvAEID, someAcousticEnvLR, someAcousticEnvERLR]



    for ts in range(startTs, endTs, 320):

    for ts in range(startTs, endTs, 320):

        pidata = dict()

        pidata = dict()

        pidata["SCENE_ORIENTATION"] = someOrientation()

        pidata["SCENE_ORIENTATION"] = someOrientation()
@@ -240,9 +257,7 @@ def generatePiData(startTs: int, endTs: int) -> dict: 
        pidata["DYNAMIC_AUDIO_SUPPRESSION_REQUEST"] = someDAS()

        pidata["DYNAMIC_AUDIO_SUPPRESSION_REQUEST"] = someDAS()

        pidata["AUDIO_DESCRIPTION"] = [someDesc() for n in range(random.randint(1, 5))]

        pidata["AUDIO_DESCRIPTION"] = [someDesc() for n in range(random.randint(1, 5))]

        pidata["DIEGETIC_TYPE"] = someDIG()

        pidata["DIEGETIC_TYPE"] = someDIG()

        pidata["ACOUSTIC_ENVIRONMENT"] = ACOUSTIC_ENVIRONMENT(

        pidata["ACOUSTIC_ENVIRONMENT"] = random.choice(someAcousticEnvList)()

            aeid=random.randint(0, 127)

        )

        data[str(ts)] = pidata

        data[str(ts)] = pidata

    return data

    return data
@@ -312,6 +327,12 @@ def isEqualAcousticEnv(ref: ACOUSTIC_ENVIRONMENT, dut: ACOUSTIC_ENVIRONMENT): 
    ), "Acoustic Env PI Data mismatch in len(abscoeff)"

    ), "Acoustic Env PI Data mismatch in len(abscoeff)"

    for r, d in zip(ref.rt60, dut.rt60):

    for r, d in zip(ref.rt60, dut.rt60):

        assert r == d, f"Acoustic Env PI Data mismatch in rt60 {r} != {d}"

        assert r == d, f"Acoustic Env PI Data mismatch in rt60 {r} != {d}"

    for r, d in zip(ref.dsr, dut.dsr):

        assert r == d, f"Acoustic Env PI Data mismatch in dsr {r} != {d}"

    for r, d in zip(ref.dim, dut.dim):

        assert r == d, f"Acoustic Env PI Data mismatch in dim {r} != {d}"

    for r, d in zip(ref.abscoeff, dut.abscoeff):

        assert r == d, f"Acoustic Env PI Data mismatch in abscoeff {r} != {d}"









def isEqualAudioFocus(ref: AUDIO_FOCUS, dut: AUDIO_FOCUS):

def isEqualAudioFocus(ref: AUDIO_FOCUS, dut: AUDIO_FOCUS):