compare two numpy arrays

cmp_np_arrays.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
{Description}
"""

from __future__ import print_function
import os
import sys
import argparse
import numpy as np

def get_args():
    parser = argparse.ArgumentParser(
        description=__doc__,
        formatter_class=argparse.RawDescriptionHelpFormatter)
    parser.add_argument('infile1', help="Input file",
                        type=str)
    parser.add_argument('infile2', help="Input file",
                        type=str)
    return parser.parse_args()


def main():
    args = get_args()
    # print(args)
    arr1 = np.loadtxt(args.infile1)
    arr2 = np.loadtxt(args.infile2)
    np.testing.assert_allclose(arr1, arr2, rtol=1e-3, atol=1e-5)
    print('check pass!')
    return 0


if __name__ == '__main__':
    sys.exit(main())