2974

2025-03-26 11:58:36 +08:00
parent a09362480a
commit 93f367dee9
3 changed files with 108 additions and 0 deletions
--- a/tests/test_2974.cpp
+++ b/tests/test_2974.cpp
@@ -0,0 +1,38 @@
+#include <gtest/gtest.h>
+#include <solution/2974.h>
+
+TEST(NumberGameTest, Test1)
+{
+    int nums[] = {5, 4, 2, 3};
+    int numsSize = sizeof(nums) / sizeof(nums[0]);
+    int returnSize = 0;
+
+    int *result = numberGame(nums, numsSize, &returnSize);
+
+    int expected[] = {3, 2, 5, 4};
+    ASSERT_EQ(returnSize, 4);
+    for (int i = 0; i < returnSize; ++i)
+    {
+        EXPECT_EQ(result[i], expected[i]);
+    }
+
+    free(result);
+}
+
+TEST(NumberGameTest, Test2)
+{
+    int nums[] = {2, 5};
+    int numsSize = sizeof(nums) / sizeof(nums[0]);
+    int returnSize = 0;
+
+    int *result = numberGame(nums, numsSize, &returnSize);
+
+    int expected[] = {5, 2};
+    ASSERT_EQ(returnSize, 2);
+    for (int i = 0; i < returnSize; ++i)
+    {
+        EXPECT_EQ(result[i], expected[i]);
+    }
+
+    free(result);
+}