tweaked test after last fix

IntelLabs · Aug 19, 2024 · 374a3ec · 374a3ec
1 parent 1f8186b
commit 374a3ec
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/python/TestDescriptors.py b/tests/python/TestDescriptors.py
@@ -298,7 +298,6 @@ def test_AddSetAndInsertBatch(self):
         blobs.append(desc_blob)
 
         response, img_array = db.query(trans, blobs)
-        print(response)
         self.assertEqual(response[0]["AddDescriptor"]["status"], 0)
 
         # now try to get those same descriptors back
@@ -313,7 +312,7 @@ def test_AddSetAndInsertBatch(self):
         blobs = []
         trans.append(query)
         response, img_array = db.query(trans, blobs)
-        #print(response)
+        self.assertEqual(response[0]["FindDescriptor"]["returned"],10)
 
         self.disconnect(db)