Add async test for classifier

lukereichold · Jul 25, 2020 · 1f569dd · 1f569dd
1 parent 3b48f08
commit 1f569dd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Tests/VisualActionKitTests/VisualActionKitTests.swift b/Tests/VisualActionKitTests/VisualActionKitTests.swift
@@ -8,10 +8,16 @@ final class VisualActionKitTests: XCTestCase {
         let url = Bundle.module.url(forResource: "writing", withExtension: "mp4")!
         let asset = AVAsset(url: url)
 
+        let expectation = self.expectation(description: "Writing Clip")
+        var actualPredictions: Classifier.Predictions?
+
         try! Classifier.shared.classify(asset) { predictions in
-            debugPrint(predictions)
+            actualPredictions = predictions
+            expectation.fulfill()
         }
 
+        waitForExpectations(timeout: 60, handler: nil)
+        XCTAssertEqual(actualPredictions?.first?.classLabel, "writing")
     }
 
     func testNormalizedColor() {