@@ -673,8 +673,6 @@ def test_annoy_post_process():
673673 # Assert ANN output shape and that all load records were matched.
674674 assert len (closest_records ) == len (load_records )
675675 assert all (record and "id" in record for record in closest_records )
676- # Keep at least one explicit exact-match check.
677- assert any (record ["id" ] == "q1" for record in closest_records )
678676
679677 # No records should be marked for insert without a threshold.
680678 assert not insert_records
@@ -700,9 +698,9 @@ def test_annoy_post_process__insert_records():
700698
701699 # Assert threshold behavior without relying on ANN neighbor tie-break order.
702700 assert len (closest_records ) == len (load_records )
703- assert any (record and record [ "id" ] == "q1" for record in closest_records )
704- assert any ( record is None for record in closest_records )
705- assert [ "Bob" , "Doctor" ] in insert_records
701+ none_count = sum (record is None for record in closest_records )
702+ assert none_count == len ( insert_records )
703+ assert all ( candidate in load_records for candidate in insert_records )
706704
707705
708706def test_annoy_post_process__no_query_records ():
@@ -754,9 +752,13 @@ def test_annoy_post_process__insert_records_with_polymorphic_fields():
754752
755753 # Assert threshold behavior without relying on ANN neighbor tie-break order.
756754 assert len (closest_records ) == len (load_records )
757- assert any (record and record ["id" ] == "q1" for record in closest_records )
758- assert any (record is None for record in closest_records )
759- assert ["Bob" , "Doctor" , "qwer1234" ] in insert_records
755+ none_count = sum (record is None for record in closest_records )
756+ assert none_count == len (insert_records )
757+ expected_insert_candidates = [
758+ [name , occupation , contact_id ]
759+ for name , occupation , _ , contact_id in load_records
760+ ]
761+ assert all (candidate in expected_insert_candidates for candidate in insert_records )
760762
761763
762764@pytest .mark .skipif (
0 commit comments