Changed error message

BinghamtonRover · Nov 26, 2024 · 4a08350 · 4a08350
1 parent 96dfba6
commit 4a08350
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/bin/analyze.py b/bin/analyze.py
@@ -6,10 +6,14 @@ def main():
   analyzer = ImageAnalyzer()
 
   no_mallet = cv2.imread("no_mallet.jpg")
-  assert analyzer.has_mallet(no_mallet) == False, "Found mallet when there is no mallet"
+  if analyzer.has_mallet(no_mallet):
+    print("Error: Found mallet when there is no mallet")
+    return
 
   yes_mallet = cv2.imread("yes_Mallet.jpg")
-  assert analyzer.has_mallet(yes_mallet) == True, "Could not detect mallet"
+  if not analyzer.has_mallet(yes_mallet):
+    print("Error: Could not detect mallet")
+    return
 
 
 main()