Function compare

test/python/utilities.py:16–31

Compare an array with a reference data file, or generate the reference file if it does not exist.

(data, reference_file, rtol=1e-8, atol=1e-12)

        return yaml_parser.load(stream)

def compare(data, reference_file, rtol=1e-8, atol=1e-12):
    """
    Compare an array with a reference data file, or generate the reference
    file if it does not exist.
    """
    data = np.array(data)
    if Path(reference_file).is_file():
        # Compare with existing output file
        ref = np.genfromtxt(reference_file)
        assert data.shape == ref.shape
        for i in range(ref.shape[0]):
            assert ref[i] == approx(data[i], rel=rtol, abs=atol)
    else:
        # Generate the output file for the first time
        warnings.warn('Generating test data file:' + Path(reference_file).resolve())
        np.savetxt(reference_file, data, fmt='%.10e')

def compareProfiles(reference, sample, rtol=1e-5, atol=1e-12, xtol=1e-5):
    """