cleanup and more sane if clause (32f693ad) · Commits · IVAS Codec Public Collaboration / IVAS Codec

tests/test_be_for_ext_outputs.py

+6 −4

Original line number	Diff line number	Diff line
		import pytest
		import pathlib
		import itertools
		from filecmp import cmp
		from tempfile import TemporaryDirectory
		from .conftest import EncoderFrontend, DecoderFrontend
		@@ -20,6 +19,8 @@ INPUT_FORMATS = [
		"HOA2",
		"HOA3",
		]
		MC_FORMATS = INPUT_FORMATS[1:6]
		SBA_FORMATS = INPUT_FORMATS[6:]

		INPUT_FILES = [
		"stvST48c_cut.wav",
		@@ -33,7 +34,6 @@ INPUT_FILES = [
		"stv3OA48c_cut.wav",
		]

		# TODO: add higher BRs
		BITRATES = [13200, 16400, 24400, 32000, 48000, 64000, 80000, 96000, 128000, 160000, 192000, 256000, 384000, 512000]


		@@ -41,15 +41,17 @@ def get_options(inp_format):
		options = list()
		if inp_format == "stereo":
		options.append("-stereo")
		elif "_" in inp_format:
		elif inp_format in MC_FORMATS:
		# this indicates MC mode
		options.extend(["-mc", inp_format])
		else:
		elif inp_format in SBA_FORMATS:
		try:
		ambi_order = int(inp_format[-1])
		except ValueError:
		ambi_order = 1
		options.extend(["-sba", str(ambi_order)])
		else:
		assert f"Can't handle input format {inp_format}"

		return options