使用 pytest 测试数据是否匹配到小数点后 4 位，并使用它发布报告

Question

我有 dataframe 如上所述，现在我必须将购买与 new_perc_buy（最多 4 位小数）相匹配，同样地与 new_perc_sell（最多 4 位小数）一起出售，如果它不符合决定的条件，它就会失败，但出于报告和比较的目的，我想使用pytest不知道怎么实现pytest有assert方法吗？

Answer 1

您可以使用 decimal.Decimal 将数字表示为特定的小数位数，然后简单地在 pytest 中断言。

import decimal

four_decimal= decimal.Decimal('0.0001')

def test():
    # get your data from database or something, if they are currently
    # representing as string, do not convert to float first but use
    # decimal.Decimal right away

    # just to show how to get the specific number of decimal places
    # usually you dont want to go float->decimal if you dont have to
    d = decimal.Decimal(1/3.0).quantize(four_decimal)
    d2 = decimal.Decimal("3214143214214321.3134").quantize(four_decimal)
    d3 = decimal.Decimal(3214143214214321/3).quantize(four_decimal)
    d4 = d2 - d - d3 + d3 + d
    # run with -s to see what's going on
    print(d)
    print(d2)
    print(d3)
    print(d4)
    # and now just assert
    assert d != d2
    assert d2 != 1/3.0
    assert d2 == d4

请注意， quantize方法也可以采用参数rounding来指定如何舍入到小数点后 4 位。