PYTHON-5121 - Use canonical Extended JSON for BSON binary vector spec tests

Author: NoahStapp

test/bson_binary_vector/float32.json

@@ -32,7 +32,7 @@
     {
       "description": "Infinity Vector FLOAT32",
       "valid": true,
-      "vector": ["-inf", 0.0, "inf"],
+      "vector": [{"$numberDouble": "-Infinity"}, 0.0, {"$numberDouble": "Infinity"} ],
       "dtype_hex": "0x27",
       "dtype_alias": "FLOAT32",
       "padding": 0,

test/test_bson_binary_vector.py

@@ -27,6 +27,15 @@
 _TEST_PATH = Path(__file__).parent / "bson_binary_vector"
 
 
+def convert_extended_json(vector) -> float:
+    if isinstance(vector, dict) and "$numberDouble" in vector:
+        if vector["$numberDouble"] == "Infinity":
+            return float("inf")
+        elif vector["$numberDouble"] == "-Infinity":
+            return float("-inf")
+    return float(vector)
+
+
 class TestBSONBinaryVector(unittest.TestCase):
     """Runs Binary Vector subtype tests.
 
@@ -62,9 +71,9 @@ def run_test(self):
                 cB_exp = binascii.unhexlify(canonical_bson_exp.encode("utf8"))
                 decoded_doc = decode(cB_exp)
                 binary_obs = decoded_doc[test_key]
-                # Handle special float cases like '-inf'
+                # Handle special extended JSON cases like 'Infinity'
                 if dtype_exp in [BinaryVectorDtype.FLOAT32]:
-                    vector_exp = [float(x) for x in vector_exp]
+                    vector_exp = [convert_extended_json(x) for x in vector_exp]
 
                 # Test round-tripping canonical bson.
                 self.assertEqual(encode(decoded_doc), cB_exp, description)