fix: ignore invalid testcase output

j178 · Nov 9, 2024 · b59bc62 · b59bc62
1 parent f9a820b
commit b59bc62
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/lang/base.go b/lang/base.go
@@ -459,10 +459,17 @@ func (l baseLang) generateTestCasesContent(q *leetcode.QuestionData) string {
 	// Assume all questions output are single.
 	var tc TestCases
 	for i := 0; i < len(cases) && i/argsNum < len(outputs); i += argsNum {
+		input := cases[i : i+argsNum]
+		output := outputs[i/argsNum]
+
+		if err := checkOutput(q, input, output); err != nil {
+			log.Debug("ignore invalid testcase output", "output", output)
+			output = ""
+		}
 		tc.AddCase(
 			TestCase{
-				Input:  cases[i : i+argsNum],
-				Output: outputs[i/argsNum],
+				Input:  input,
+				Output: output,
 			},
 		)
 	}