Add helper function for testing

getbouncer · Nov 1, 2019 · e8b3ec4 · e8b3ec4
1 parent bad1702
commit e8b3ec4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/CardScan/Classes/Ocr.swift b/CardScan/Classes/Ocr.swift
@@ -42,6 +42,13 @@ public class Ocr {
                                            width: xMax - xMin, height: yMax - yMin)
     }
 
+    // used just for testing
+    @available(iOS 11.2, *)
+    public static func updateDetectionModel(resourceName: String) {
+        FindFourOcr.detectModel = nil
+        FindFourOcr.findFourResource = resourceName
+    }
+
     public func userCancelled() {
         self.scanStats.success = false
         self.scanStats.endTime = Date()