我可以在单元测试中比较2个浮点的相似性，而不是相等性吗？

Question

我上面有这个测试。 我的问题是，在训练完机器学习模型之后，每次的结果都会有所不同。

因此，当我的Flask-api中类似时，我想断言TRUE。

ChangeToSurvive是float ，所以有时是29.2 ，在训练后再次更改为30.5或其他数字。

def test_by_name(self):
    post_data = {
        'name': 'Andre'
    }
    resp = self.app.post('/survivals',
                         data=json.dumps(post_data),
                         content_type='application/json')
    self.assertEqual(resp.status_code, 200)
    self.assertEqual(resp.content_type, 'application/json')

    content = json.loads(resp.get_data(as_text=True))
    size = len(content['Passengers'])
    self.assertEqual(size, 2)

    self.maxDiff = None 

    expected = {
"Passengers": [
    {
        "SibSp": 1,
        "Sex": "0",
        "PassengerId": 925,
        "Survived": 1,
        "Parch": 2,
        "Age": 1,
        "Name": "Johnston, Mrs. Andrew G (Elizabeth Lily\" Watson)\"",
        "ChangeToSurvive": 74.7,
        "Embarked": "0"
    },
    {
        "SibSp": 0,
        "Sex": "1",
        "PassengerId": 1096,
        "Survived": 0,
        "Parch": 0,
        "Age": 1,
        "Name": "Andrew, Mr. Frank Thomas",
        "ChangeToSurvive": 29.2,
        "Embarked": "0"
    }
]
}
    self.assertEqual(content, expected)   # each train in a model the ChangeToSurvive can a little diferrent

Answer 1

为此， unittest具有assertAlmostEqual方法：

actual_value = # extract float value from content
expected_value = # extract float value from expected
delta = 1.7
self.assertAlmostEqual(actual_value, expected_value, delta = delta)

您需要对要验证的字典中的所有值重复此操作。 最好用测试实用程序方法包装它。