GTest unit testing framework

Currently covers only small set of functionality.

1 files changed, 78 insertions, 0 deletions

diff --git a/tests/gtests/testing/testing.h b/tests/gtests/testing/testing.h
new file mode 100644
index 00000000000..9083654f1d5
--- /dev/null
+++ b/tests/gtests/testing/testing.h
@@ -0,0 +1,78 @@
+#ifndef __BLENDER_TESTING_H__
+#define __BLENDER_TESTING_H__
+
+#include "glog/logging.h"
+#include "gflags/gflags.h"
+#include "gtest/gtest.h"
+
+#define EXPECT_V3_NEAR(a, b, eps) \
+  { \
+    EXPECT_NEAR(a[0], b[0], eps); \
+    EXPECT_NEAR(a[1], b[1], eps); \
+    EXPECT_NEAR(a[2], b[2], eps); \
+  } (void) 0
+
+#define EXPECT_MATRIX_NEAR(a, b, tolerance) \
+do { \
+  bool dims_match = (a.rows() == b.rows()) && (a.cols() == b.cols()); \
+  EXPECT_EQ(a.rows(), b.rows()) << "Matrix rows don't match."; \
+  EXPECT_EQ(a.cols(), b.cols()) << "Matrix cols don't match."; \
+  if (dims_match) { \
+    for (int r = 0; r < a.rows(); ++r) { \
+      for (int c = 0; c < a.cols(); ++c) { \
+        EXPECT_NEAR(a(r, c), b(r, c), tolerance) \
+          << "r=" << r << ", c=" << c << "."; \
+      } \
+    } \
+  } \
+} while(false);
+
+#define EXPECT_MATRIX_NEAR_ZERO(a, tolerance) \
+do { \
+  for (int r = 0; r < a.rows(); ++r) { \
+    for (int c = 0; c < a.cols(); ++c) { \
+      EXPECT_NEAR(0.0, a(r, c), tolerance) \
+        << "r=" << r << ", c=" << c << "."; \
+    } \
+  } \
+} while(false);
+
+#define EXPECT_MATRIX_EQ(a, b) \
+do { \
+  bool dims_match = (a.rows() == b.rows()) && (a.cols() == b.cols()); \
+  EXPECT_EQ(a.rows(), b.rows()) << "Matrix rows don't match."; \
+  EXPECT_EQ(a.cols(), b.cols()) << "Matrix cols don't match."; \
+  if (dims_match) { \
+    for (int r = 0; r < a.rows(); ++r) { \
+      for (int c = 0; c < a.cols(); ++c) { \
+        EXPECT_EQ(a(r, c), b(r, c)) \
+          << "r=" << r << ", c=" << c << "."; \
+      } \
+    } \
+  } \
+} while(false);
+
+// Check that sin(angle(a, b)) < tolerance.
+#define EXPECT_MATRIX_PROP(a, b, tolerance) \
+do { \
+  bool dims_match = (a.rows() == b.rows()) && (a.cols() == b.cols()); \
+  EXPECT_EQ(a.rows(), b.rows()) << "Matrix rows don't match."; \
+  EXPECT_EQ(a.cols(), b.cols()) << "Matrix cols don't match."; \
+  if (dims_match) { \
+    double c = CosinusBetweenMatrices(a, b); \
+    if (c * c < 1) { \
+      double s = sqrt(1 - c * c); \
+      EXPECT_NEAR(0, s, tolerance); \
+    } \
+  } \
+} while(false);
+
+#ifdef LIBMV_NUMERIC_NUMERIC_H
+template<class TMat>
+double CosinusBetweenMatrices(const TMat &a, const TMat &b) {
+  return (a.array() * b.array()).sum() /
+      libmv::FrobeniusNorm(a) / libmv::FrobeniusNorm(b);
+}
+#endif
+
+#endif  // __BLENDER_TESTING_H__