Fix test normalization

argmaxinc · May 23, 2024 · 524a10d · 524a10d
1 parent b847da6
commit 524a10d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Tests/WhisperKitTests/UnitTests.swift b/Tests/WhisperKitTests/UnitTests.swift
@@ -1110,11 +1110,11 @@ final class UnitTests: XCTestCase {
 
         // Select few sentences to compare at VAD border
         // TODO: test that WER is in acceptable range
-        XCTAssertTrue(testResult.text.normalized.contains("and then I would kind of do this"), "Expected text not found in \(testResult.text.normalized)")
-        XCTAssertTrue(chunkedResult.text.normalized.contains("and then I would kind of do this"), "Expected text not found in \(chunkedResult.text.normalized)")
+        XCTAssertTrue(testResult.text.normalized.contains("and then I would kind of do this".normalized), "Expected text not found in \(testResult.text.normalized)")
+        XCTAssertTrue(chunkedResult.text.normalized.contains("and then I would kind of do this".normalized), "Expected text not found in \(chunkedResult.text.normalized)")
 
-        XCTAssertTrue(testResult.text.normalized.contains("And that would happen every single paper"), "Expected text not found in \(testResult.text.normalized)")
-        XCTAssertTrue(chunkedResult.text.normalized.contains("And that would happen every single paper"), "Expected text not found in \(chunkedResult.text.normalized)")
+        XCTAssertTrue(testResult.text.normalized.contains("And that would happen every single paper".normalized), "Expected text not found in \(testResult.text.normalized)")
+        XCTAssertTrue(chunkedResult.text.normalized.contains("And that would happen every single paper".normalized), "Expected text not found in \(chunkedResult.text.normalized)")
 
         XCTAssertTrue(testResult.text.normalized.contains("But then came my 90 page senior".normalized), "Expected text not found in \(testResult.text.normalized)")
         XCTAssertTrue(chunkedResult.text.normalized.contains("But then came my 90 page senior".normalized), "Expected text not found in \(chunkedResult.text.normalized)")