diff --git a/bob/bio/face/test/test_extractors.py b/bob/bio/face/test/test_extractors.py index e1dfc0762f34f152a17ee360ef7fa880f0859180..97fee729e69b38eb216678e5d1b67cacd04753dd 100644 --- a/bob/bio/face/test/test_extractors.py +++ b/bob/bio/face/test/test_extractors.py @@ -33,14 +33,14 @@ import pkg_resources regenerate_refs = False -def _compare(data, reference, write_function = bob.bio.base.save, read_function = bob.bio.base.load): +def _compare(data, reference, write_function = bob.bio.base.save, read_function = bob.bio.base.load, atol = 1e-5, rtol = 1e-8): # write reference? if regenerate_refs: write_function(data, reference) # compare reference reference = read_function(reference) - assert numpy.allclose(data, reference, atol=1e-5) + assert numpy.allclose(data, reference, atol=atol, rtol=rtol) def _data(): return bob.bio.base.load(pkg_resources.resource_filename('bob.bio.face.test', 'data/cropped.hdf5')) diff --git a/bob/bio/face/test/test_preprocessors.py b/bob/bio/face/test/test_preprocessors.py index 1495394788bf7d8bc32d648bea1526fbc26bf3ec..815b74bfffd0487bfcedc0733abd9d29b0b3ce8c 100644 --- a/bob/bio/face/test/test_preprocessors.py +++ b/bob/bio/face/test/test_preprocessors.py @@ -33,14 +33,14 @@ import bob.bio.face import bob.db.verification.utils -def _compare(data, reference, write_function = bob.bio.base.save, read_function = bob.bio.base.load): +def _compare(data, reference, write_function = bob.bio.base.save, read_function = bob.bio.base.load, atol = 1e-5, rtol = 1e-8): # write reference? if regenerate_refs: write_function(data, reference) # compare reference reference = read_function(reference) - assert numpy.allclose(data, reference, atol=1e-5) + assert numpy.allclose(data, reference, atol=atol, rtol=rtol) def _image():