fix print tests for Rot3 and Pose3

release/4.3a0
Varun Agrawal 2020-06-17 14:55:52 -05:00
parent 944a437272
commit f7141f4333
2 changed files with 9 additions and 4 deletions

View File

@ -839,7 +839,8 @@ TEST( Pose3, stream)
Pose3 T; Pose3 T;
std::ostringstream os; std::ostringstream os;
os << T; os << T;
EXPECT(os.str() == "\n|1, 0, 0|\n|0, 1, 0|\n|0, 0, 1|\n\n[0, 0, 0]';\n"); string expected = "[\n\t1, 0, 0;\n\t0, 1, 0;\n\t0, 0, 1\n]\n\n[0, 0, 0]';\n";
EXPECT(os.str() == expected);
} }
//****************************************************************************** //******************************************************************************
@ -1009,19 +1010,22 @@ TEST(Pose3, print) {
std::stringstream expected; std::stringstream expected;
Point3 translation(1, 2, 3); Point3 translation(1, 2, 3);
// Add expected rotation
expected << "R:\n[\n\t1, 0, 0;\n\t0, 1, 0;\n\t0, 0, 1\n]\n";
#ifdef GTSAM_TYPEDEF_POINTS_TO_VECTORS #ifdef GTSAM_TYPEDEF_POINTS_TO_VECTORS
expected << "1\n" expected << "1\n"
"2\n" "2\n"
"3;\n"; "3;\n";
#else #else
expected << '[' << translation.x() << ", " << translation.y() << ", " << translation.z() << "]\';"; expected << '[' << translation.x() << ", " << translation.y() << ", " << translation.z() << "]\';\n";
#endif #endif
// reset cout to the original stream // reset cout to the original stream
std::cout.rdbuf(oldbuf); std::cout.rdbuf(oldbuf);
// Get substring corresponding to translation part // Get substring corresponding to translation part
std::string actual = redirectStream.str().substr(38, 11); std::string actual = redirectStream.str();
CHECK_EQUAL(expected.str(), actual); CHECK_EQUAL(expected.str(), actual);
} }

View File

@ -608,7 +608,8 @@ TEST( Rot3, stream)
Rot3 R; Rot3 R;
std::ostringstream os; std::ostringstream os;
os << R; os << R;
EXPECT(os.str() == "\n|1, 0, 0|\n|0, 1, 0|\n|0, 0, 1|\n"); string expected = "[\n\t1, 0, 0;\n\t0, 1, 0;\n\t0, 0, 1\n]\n";
EXPECT(os.str() == expected);
} }
/* ************************************************************************* */ /* ************************************************************************* */