Google Test中的自定义EXPECT_NEAR宏

时间:2011-08-19 11:13:20

标签: unit-testing macros opencv googletest

范围:使用Google Test和OpenCV。

我想测试我的Vec3f等于另一个Vec3fVec3f是OpenCV中尺寸为3且类型为float的向量。定义了== - 运算符,因此EXPECT_EQ(Vec3f(), Vec3f())有效。

但是因为它们是花车,我想使用EXPECT_NEAR(float a, float b, float delta)宏。我可以做什么,以便像EXPECT_NEAR(vec_a, vec_b, float delta)一样使用它?

目前我循环遍历向量的每个元素并在那里进行EXPECT_NEAR。

这可能是相关的:Convenient method in GoogleTest for a double comparison of not equal?

2 个答案:

答案 0 :(得分:7)

您可以使用Google Mock中的Pointwise()匹配器。将它与自定义匹配器结合使用,该匹配器检查两个参数是否接近:

#include <tr1/tuple>
#include <gmock/gmock.h>

using std::tr1::get;
using testing::Pointwise;

MATCHER_P(NearWithPrecision, precision, "") {
  return abs(get<0>(arg) - get<1>(arg)) < precision;
}

TEST(FooTest, ArraysNear) {
  EXPECT_THAT(result_array, Pointwise(NearWithPrecision(0.1), expected_array));
}

答案 1 :(得分:3)

你基本上做的是正确的事情。但是,我会使用自定义断言函数,如:

::testing::AssertionResult AreAllElementsInVectorNear(const Vec3f& a, const Vect3f& b, float delta) {
  if ([MAGIC])
    return ::testing::AssertionSuccess();
  else
    return ::testing::AssertionFailure() << "Vectors differ by more than " << delta;
}

MAGIC会将您的代码包括在内,例如比较两个向量是否具有相同的大小,然后迭代所有元素并相互检查相同索引处的元素是否相差不超过delta。请注意,代码假定&lt;&lt; Vec3f提供了运算符。

然后使用该功能:

EXPECT_TRUE(AreAllElementsInVectorNear(a, b, 0.1))

如果期望失败,输出可能是:

Value of: AreAllElementsInVectorNear(a, b, 0.1)
  Actual: false (Vectors differ by more then 0.1)
Expected: true