1 files changed, 3 insertions, 4 deletions
diff --git a/test/src/TestAPI.kt b/test/src/TestAPI.kt
index 6f4e34c2..cc09f001 100644
--- a/test/src/TestAPI.kt
+++ b/test/src/TestAPI.kt
@@ -51,13 +51,12 @@ public fun verifyModel(vararg files: String, verifier: (DocumentationModule) ->
     Disposer.dispose(environment)
 }
 
-public fun verifyOutput(path: String, outputGenerator: (DocumentationModule, StringBuilder) -> Unit) {
+public fun verifyOutput(path: String, outputExtension: String, outputGenerator: (DocumentationModule, StringBuilder) -> Unit) {
     verifyModel(path) {
         val output = StringBuilder()
         outputGenerator(it, output)
-        val trimmedOutput = output.toString().split('\n').map { it.trimTrailing() }.join("\n")
-        val expectedOutput = File(path.replace(".kt", ".md")).readText()
-        assertEquals(expectedOutput.trimTrailing(), trimmedOutput)
+        val expectedOutput = File(path.replace(".kt", outputExtension)).readText()
+        assertEquals(expectedOutput, output.toString())
     }
 }