Commit 015a779b authored by Jan Kiene's avatar Jan Kiene
Browse files

fix printout and use integer data type

parent 1ec57e1f
Loading
Loading
Loading
Loading
Loading
+5 −3
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ sys.path.append(os.path.join(os.path.dirname(THIS_PATH), "../scripts"))

import pyaudio3dtools
import pyivastest
import numpy as np


def cmp_pcm(file1, file2, out_config, fs) -> (int, str):
@@ -26,8 +27,8 @@ def cmp_pcm(file1, file2, out_config, fs) -> (int, str):
        out_config_in_file_names = out_config
        nchannels = pyivastest.constants.OC_TO_NCHANNELS[out_config.upper()]

    s1, _ = pyaudio3dtools.audiofile.readfile(file1, nchannels, fs)
    s2, _ = pyaudio3dtools.audiofile.readfile(file2, nchannels, fs)
    s1, _ = pyaudio3dtools.audiofile.readfile(file1, nchannels, fs, outdtype=np.int16)
    s2, _ = pyaudio3dtools.audiofile.readfile(file2, nchannels, fs, outdtype=np.int16)

    if s1.shape != s2.shape:
        print(
@@ -54,5 +55,6 @@ if __name__ == "__main__":
    parser.add_argument("-s", "--sampling_rate", type=int, default=48000, dest="fs")
    args = parser.parse_args()

    result, _ = cmp_pcm(**vars(args))
    result, msg = cmp_pcm(**vars(args))
    print(msg)
    sys.exit(result)