python 比较两个浮点或者定点数据的二进制文件是否一致

Posted 2021-05-09

import os
import zbin
import numpy as np


def main(dir1, dir2):
	total_err = 0
	idx  = 0
	files = [f for f in os.listdir(dir1) if os.path.splitext(f)[-1] == '.bin']
	total = len(files)
	err_f = []
	for f in files:
		f_1 = os.path.join(dir1, f)
		f_2 = os.path.join(dir2, f)
		
		f_1_bin = zbin.read(f_1, 32, num_type='float')
		f_1_bin = np.array(f_1_bin)

		f_2_bin = zbin.read(f_2, 32, num_type='float')
		f_2_bin = np.array(f_2_bin)

		err = np.max(np.abs(f_1_bin - f_2_bin))
		if err != 0:
			err_f.append(f)
		total_err += err

		pos = int((idx / total) * 100) + 1
		print("\rSuccessfully cmp {}{}{}/{}".format(pos*'>', (100-pos)*'=', pos, 100), end='', flush=True)
		idx += 1

	print('\n')
	print("Result: %d" % total_err)
	for f in err_f:
		print(f)


if __name__ == '__main__':
	main('float_little_old', 'float_little')