Commit 6042b4af authored by Vladimir Malenovsky's avatar Vladimir Malenovsky
Browse files

fix standalone function

parent 4514cc06
Loading
Loading
Loading
Loading
Loading
+9 −15
Original line number Diff line number Diff line
@@ -126,20 +126,14 @@ def cmp_stats_files(

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("ref_file", type=str)
    parser.add_argument("cut_file", type=str)
    parser.add_argument("--data_type", type=str, default="int16", dest="dtype")
    parser.add_argument(
        "--nsamples_per_frame", type=str, default=960, dest="nsamples_per_frame"
    )
    parser.add_argument("--len_check", type=int, default=0, dest="len_check")
    parser.add_argument("--min_diff_thr", type=float, default=0.0, dest="min_diff_thr")
    
    parser.add_argument("ref_stats_file", type=str)
    parser.add_argument("dut_stats_file", type=str)
    parser.add_argument("--min_enc_file_length_diff", type=float, default=0.1, dest="min_enc_file_length_diff")
    parser.add_argument("--min_enc_stats_diff", type=float, default=0.1, dest="min_enc_stats_diff")
    args = parser.parse_args()

    # convert dtype from str
    if isinstance(args.dtype, str):
        args.dtype = np.dtype(getattr(np, args.dtype))
    enc_test_result, enc_test_result_msg = cmp_stats_files(**vars(args))
    print(enc_test_result_msg)
    
    result, msg = cmp_bin_files(**vars(args))
    print(msg)
    sys.exit(result)
    sys.exit(enc_test_result)