s0006

2022-11-02 20:19:54 +08:00
parent a12d393198
commit 83c66fc158
3 changed files with 98 additions and 0 deletions
--- a/tests/s0006_zigzag_conversion.cpp
+++ b/tests/s0006_zigzag_conversion.cpp
@@ -0,0 +1,27 @@
+#include "s0006_zigzag_conversion.hpp"
+
+#include <gtest/gtest.h>
+
+TEST(Problem6, Case1) {
+  string s("PAYPALISHIRING");
+  int rows = 3;
+  string r("PAHNAPLSIIGYIR");
+  Solution solution;
+  EXPECT_TRUE(solution.convert(s, rows) == r);
+}
+
+TEST(Problem6, Case2) {
+  string s("PAYPALISHIRING");
+  int rows = 4;
+  string r("PINALSIGYAHRPI");
+  Solution solution;
+  EXPECT_TRUE(solution.convert(s, rows) == r);
+}
+
+TEST(Problem6, Case3) {
+  string s("AB");
+  int rows = 4;
+  string r("AB");
+  Solution solution;
+  EXPECT_TRUE(solution.convert(s, rows) == r);
+}