qvec: allow gtest to print

ericwa · Sep 19, 2024 · f5d105b · f5d105b
1 parent ed62ec1
commit f5d105b
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/include/common/qvec.hh b/include/common/qvec.hh
@@ -296,6 +296,12 @@ public:
     constexpr auto end() const { return v.end(); }
     constexpr auto cbegin() const { return v.cbegin(); }
     constexpr auto cend() const { return v.cend(); }
+
+    // gtest support
+    friend std::ostream& operator<<(std::ostream& os, const qvec& point) {
+        os << fmt::format("{}", point);
+        return os;
+    }
 };
 
 // Fmt support

diff --git a/tests/test_light.cc b/tests/test_light.cc
@@ -717,6 +717,20 @@ TEST(mathlib, qvecConstructorExtra)
     EXPECT_EQ(2, test[1]);
 }
 
+TEST(mathlib, qvecGTestPrint)
+{
+    const qvec2f test(1, 2);
+
+    EXPECT_EQ(testing::PrintToString(test), "1 2");
+}
+
+TEST(mathlib, qvecFmtFormat)
+{
+    const qvec2f test(1, 2);
+
+    EXPECT_EQ("1 2", fmt::format("{}", test));
+}
+
 // aabb3f
 
 TEST(mathlib, aabbBasic)