Multi-line test input/output

Author: ShotaKashihara

Sources/AtCoderLibrary/Command/New/Generator/Test.swift

@@ -13,13 +13,23 @@ struct Test: Generator {
 
             func testExample() throws {
                 let cases: [TestCase] = [
-        \(problem.tests.map { """
-                    (#filePath, #line, \"\($0.input.replacingOccurrences(of: "\n", with: "\\n"))\", \"\($0.output.replacingOccurrences(of: "\n", with: "\\n"))\"),
-        """ }.joined(separator: "\n"))
+        \(problem.tests.map(Self.testCase).joined(separator: "\n"))
                 ]
                 try cases.forEach(solve)
             }
         }
         """
     }
+
+    private static func testCase(_ test: Problem.Test) -> String {
+        let inputs = test.input.split(separator: "\n")
+        let outputs = test.output.split(separator: "\n")
+        return """
+                        (#filePath, #line, \"""
+            \(inputs.map { "                \($0)" }.joined(separator: "\n"))
+                            \""", \"""
+            \(outputs.map { "                \($0)" }.joined(separator: "\n"))
+                            \"""),
+            """
+    }
 }

Sources/AtCoderLibrary/Command/New/Generator/TestLibrary.swift

@@ -1,6 +1,6 @@
 struct TestLibrary: Generator {
     let directory: String? = "Tests/TestLibrary"
-    let fileName = "Library.swift"
+    let fileName = "TestLibrary.swift"
     var source: String {
         """
         import XCTest
@@ -18,6 +18,7 @@ struct TestLibrary: Generator {
                 guard #available(macOS 10.13, *) else {
                     return
                 }
+                let expected = expected.last == "\\n" ? expected : expected + "\\n"
                 var error = ""
                 let exp = expectation(description: "")
                 let testTarget = String(describing: type(of: self)).replacingOccurrences(of: "Tests", with: "")

Tests/AtCoderLibraryTests/Command/New/Generator/TestTests.swift

@@ -7,9 +7,8 @@ final class TestTests: XCTestCase {
         let problem = Problem(
             url: "https://example.com/p_1".url,
             tests: [
-                .init(input: "input1", output: "output1"),
-                .init(input: "input2", output: "output2"),
-                .init(input: "input3", output: "output3"),
+                .init(input: "10 2\n20 3\n30 4\n", output: "5\n20\n3\n"),
+                .init(input: "199 2\n", output: "1992\n"),
             ],
             name: "",
             context: .init(contest: .empty, alphabet: "A"),
@@ -26,9 +25,20 @@ final class TestTests: XCTestCase {
 
             func testExample() throws {
                 let cases: [TestCase] = [
-                    (#filePath, #line, "input1", "output1"),
-                    (#filePath, #line, "input2", "output2"),
-                    (#filePath, #line, "input3", "output3"),
+                    (#filePath, #line, \"""
+                        10 2
+                        20 3
+                        30 4
+                        \""", \"""
+                        5
+                        20
+                        3
+                        \"""),
+                    (#filePath, #line, \"""
+                        199 2
+                        \""", \"""
+                        1992
+                        \"""),
                 ]
                 try cases.forEach(solve)
             }