PYtest: valer casi igual

Question

Cómo hacer assert almost equal con py.test de los flotadores sin tener que recurrir a algo como:

assert x - 0.00001 <= y <= x + 0.00001 

Más específicamente, será útil conocer una solución ordenada para comparar rápidamente los pares de flotador, sin desempaquetarlos:

assert (1.32, 2.4) == i_return_tuple_of_two_floats() 

Answer 1

me di cuenta de que esta pregunta le preguntó específicamente sobre py.test. py.test 3.0 incluye una función approx() (bueno, realmente clase) que es muy útil para este propósito.

import pytest 

assert 2.2 == pytest.approx(2.3) 
# fails, default is ± 2.3e-06 
assert 2.2 == pytest.approx(2.3, 0.1) 
# passes 

# also works the other way, in case you were worried: 
assert pytest.approx(2.3, 0.1) == 2.2 
# passes 

La documentación está aquí: http://doc.pytest.org/en/latest/builtin.html#pytest.approx

Answer 2

tendrá que especificar lo que es "casi" para usted:

assert abs(x-y) < 0.0001 

aplicar a tuplas (o cualquier secuencia):

def almost_equal(x,y,threshold=0.0001): 
    return abs(x-y) < threshold 

assert all(map(almost_equal, zip((1.32, 2.4), i_return_tuple_of_two_floats()) 

Answer 3

Algo así como

assert round(x-y, 5) == 0 

Eso es lo que unittest hace

Para la segunda parte

assert all(round(x-y, 5) == 0 for x,y in zip((1.32, 2.4), i_return_tuple_of_two_floats())) 

Probablemente es mejor envolver eso en una función

def tuples_of_floats_are_almost_equal(X, Y): 
    return all(round(x-y, 5) == 0 for x,y in zip(X, Y)) 

assert tuples_of_floats_are_almost_equal((1.32, 2.4), i_return_tuple_of_two_floats()) 

Answer 4

Si tiene acceso a NumPy tiene grandes funciones para la comparación de punto flotante que ya hacen comparación por pares: http://docs.scipy.org/doc/numpy-dev/reference/routines.testing.html.

A continuación, puede hacer algo como:

numpy.testing.assert_allclose(i_return_tuple_of_two_floats(), (1.32, 2.4)) 

Answer 5

Estas respuestas han existido por mucho tiempo, pero creo que el más fácil y también forma más legible es utilizar unittest por su many nice assertions sin usarlo para el estructura de prueba.

Obtener afirmaciones, ignorar resto de unittest.TestCase

(basado en this answer)

import unittest 

assertions = unittest.TestCase('__init__') 

hacen que algunas afirmaciones

x = 0.00000001 
assertions.assertAlmostEqual(x, 0) # pass 
assertions.assertEqual(x, 0) # fail 
# AssertionError: 1e-08 != 0 

Implementar la prueba de auto-desembalaje preguntas originales

Solo use * para desempaquetar su valor de retorno sin needi ng para introducir nuevos nombres.

i_return_tuple_of_two_floats = lambda: (1.32, 2.4) 
assertions.assertAlmostEqual(*i_return_tuple_of_two_floats()) # fail 
# AssertionError: 1.32 != 2.4 within 7 places 

Answer 6

Utilizaría nose.tools. Se juega bien con el corredor py.test y tienen otra igualmente útil afirma - assert_dict_equal(), assert_list_equal(), etc.

from nose.tools import assert_almost_equals 
assert_almost_equals(x, y, places=7) #default is 7