4

我正在尝试编写一个单元测试来检查工程分析的输出。我有理论值，我想对照分析对一定数量的有效数字进行检查。所以，例如：

Ixx_ther = 0.000123
Iyy_ther = 0.0123

Ixx, Iyy = getI(*args, **kwargs)

self.assertAlmostEqual(Ixx_ther, Ixx, 6)
self.assertAlmostEqual(Iyy_ther, Iyy, 4)

在这种情况下，我需要知道我要检查的数字，因为在这两种情况下将容差设置为 6 会使测试过于严格，而将其设置为 4 则过于宽松。我需要的是测试相同数量的有效数字是否相等。理想的说法是：

Ixx_ther = 1.23E-4
Iyy_ther = 1.23E-2

Ixx, Iyy = getI(*args, **kwargs)

self.assertAlmostEqual(Ixx_ther, Ixx, 2)
self.assertAlmostEqual(Iyy_ther, Iyy, 2)

并让断言语句删除指数并仅检查有效数是否相等。我想这以前已经做过了，但是我还没有找到一个内置函数来以这种方式断言相等。以前有没有人遇到过这个问题

问题

1）以前有没有人遇到过这个问题，并且知道用于工程分析的单元测试的一般指南

2）是否有内置解决方案。对于这个问题

3) 是否有人已经编写了以这种方式工作的自定义断言语句？

4

4 回答 4

4

回复：是否有内置解决方案：如果您可以numpy作为依赖项，请查看 numpy.testing。

这是一个示例（来自assert_allclose文档的逐字记录）：

>>> x = [1e-5, 1e-3, 1e-1]
>>> y = np.arccos(np.cos(x))
>>> assert_allclose(x, y, rtol=1e-5, atol=0)

编辑：为了完整起见，这里是源代码的链接：assert_allclose将实际工作转发给np.allclose。这与@Mark Ransom 的答案几乎相同（加上对数组参数和无穷大的处理）。

于 2013-10-30T17:01:48.520 回答
4

这是对我在另一个问题上留下的答案的修改。

def AlmostEqual(a, b, digits):
    epsilon = 10 ** -digits
    return abs(a/b - 1) < epsilon

b如果可以为零，这需要更多的工作。

于 2013-10-30T17:14:32.113 回答
0

也许没有回答您问题的全部范围，但这就是我编写这样一个函数的方式：

def assertAlmostEqual(arg1,arg2,tolerance=2):
    str_formatter = '{0:.' + str(tolerance) + 'e}'
    lhs = str_formatter.format(arg1).split('e')[0]
    rhs = str_formatter.format(arg2).split('e')[0]
    assert lhs == rhs

可以利用Python 的字符串格式化迷你语言将浮点数格式化为给定的方式。所以我们可以做的是强制它们以指数表示法格式化，这样，对于输入0.123，0.000123我们有：

str_formatter.format(0.123) == '1.23e-01'
str_formatter.format(0.000123) == '1.23e-04'

剩下的就是砍掉指数并断言相等。

演示：

assertAlmostEqual(0.0123,0.0001234)

assertAlmostEqual(0.123,0.0001234)

assertAlmostEqual(0.123,0.0001234,tolerance=3)
---------------------------------------------------------------------------
AssertionError                            Traceback (most recent call last)
/home/xxx/<ipython-input-83-02fbd71b2e87> in <module>()
----> 1 assertAlmostEqual(0.123,0.0001234,tolerance=3)

/home/xxx/<ipython-input-74-ae32ed74769d> in assertAlmostEqual(arg1, arg2, tolerance)
      3     lhs = str_formatter.format(arg1).split('e')[0]
      4     rhs = str_formatter.format(arg2).split('e')[0]
----> 5     assert lhs == rhs
      6 

AssertionError:

如果您不喜欢我定义的tolerance. 得到这个想法。

于 2013-10-28T20:26:46.953 回答
0

感谢roippi您的好主意，我稍微修改了您的代码：

def assertAlmostEqualSigFig(self, arg1,arg2,tolerance=2):
    if tolerance > 1: 
        tolerance -= 1
    #end

    str_formatter = '{0:.' + str(tolerance) + 'e}'
    significand_1 = float(str_formatter.format(arg1).split('e')[0])
    significand_2 = float(str_formatter.format(arg2).split('e')[0])

    exponent_1 = int(str_formatter.format(arg1).split('e')[1])
    exponent_2 = int(str_formatter.format(arg2).split('e')[1])

    self.assertEqual(significand_1, significand_2)
    self.assertEqual(exponent_1, exponent_2)

    return

我改变了一些东西

1）我检查指数和有效数字（那是最上面的抽屉词不是吗）

2）我将有效数和指数分别转换为浮点/整数。这可能不是必需的，但我更愿意将数字的相等性检查为数字而不是字符串。

3)Jim Lewis注意到我需要将容差调整一，因为 0.0123 的正确格式字符串 {0:.3e} 是 1.230E-2 而不是 0.123E-1。即，如果您想要三个有效数字，您只需要小数点后的两位数，因为小数点前的数字也很重要。

她的就是一个实现的例子

class testSigFigs(Parent_test_class):

    @unittest.expectedFailure
    def test_unequal_same_exp(self):
        self.assertAlmostEqualSigFig(0.123, 0.321, 3)

    @unittest.expectedFailure
    def test_unequal_diff_exp(self):
        self.assertAlmostEqualSigFig(0.123, 0.0321, 3)

    @unittest.expectedFailure
    def test_equal_diff_exp(self):
        self.assertAlmostEqualSigFig(0.0123, 0.123, 3)

    def test_equal_same_exp(self):
        self.assertAlmostEqualSigFig(0.123, 0.123, 3)

    def test_equal_within_tolerance(self):
        self.assertAlmostEqualSigFig(0.123, 0.124, 2)
    #end

和输出：

test_equal_diff_exp (__main__.testSigFigs) ... expected failure
test_equal_same_exp (__main__.testSigFigs) ... ok
test_equal_within_tolerance (__main__.testSigFigs) ... ok
test_unequal_diff_exp (__main__.testSigFigs) ... expected failure
test_unequal_same_exp (__main__.testSigFigs) ... expected failure

----------------------------------------------------------------------
Ran 5 tests in 0.081s

OK (expected failures=3)

谢谢你们的反馈。

于 2013-10-29T14:54:50.390 回答