pytest : assert almost equal

Question

Comment faire assert almost equal avec py.test pour les flottants sans avoir recours à quelque chose comme :

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

Plus précisément, il sera utile de connaître une solution soignée pour comparer rapidement des paires de flotteurs, sans les déballer :

assert (1.32, 2.4) == i_return_tuple_of_two_floats()

Answer 1

J'ai remarqué que cette question portait spécifiquement sur py.test. py.test 3.0 comprend un approx() (enfin, une vraie classe) qui est très utile à cet effet.

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 documentation est ici : https://docs.pytest.org/en/latest/reference/reference.html?highlight=approx#pytest.approx

Answer 2

Vous devrez préciser ce qui est "presque" pour vous :

assert abs(x-y) < 0.0001

pour s'appliquer aux tuples (ou à toute séquence) :

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

Si vous avez accès à NumPy, il dispose d'excellentes fonctions de comparaison en virgule flottante qui permettent déjà de faire des comparaisons par paire avec numpy.testing .

Alors vous pouvez faire quelque chose comme :

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

Answer 4

Ces réponses existent depuis longtemps, mais je pense que le moyen le plus simple et le plus lisible est d'utiliser unittest pour sa fonction beaucoup de belles affirmations sans l'utiliser pour la structure de test.

Obtenir les assertions, ignorer le reste de unittest.TestCase

(sur la base de cette réponse )

import unittest

assertions = unittest.TestCase('__init__')

Faites quelques affirmations

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

Mise en œuvre du test de déballage automatique des questions originales

Utilisez simplement * pour décompresser votre valeur de retour sans avoir à introduire de nouveaux noms.

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 5

Si vous voulez quelque chose qui fonctionne non seulement avec les flottants mais aussi avec les décimales, vous pouvez utiliser la fonction de python math.isclose :

    # - rel_tol=0.01` is 1% difference tolerance.
    assert math.isclose(actual_value, expected_value, rel_tol=0.01)

Docs - https://docs.python.org/3/library/math.html#math.isclose