ARROW-155 Ensure compatibility with Pandas 2.0 (#139)

Author: blink1073

bindings/python/pymongoarrow/pandas_types.py

@@ -229,6 +229,12 @@ def __arrow_array__(self, type=None):
 
         return pa.array(self.data, type=BinaryType(self.dtype.subtype))
 
+    def __eq__(self, other):
+        # Binary types do not support element-wise comparison.
+        if isinstance(other, Binary):
+            other = np.array(other, dtype=object)
+        return super().__eq__(other)
+
 
 @register_extension_dtype
 class PandasObjectId(PandasBSONDtype):

bindings/python/setup.cfg

@@ -37,7 +37,7 @@ install_requires =
     # keep versions in sync with pyproject.toml "requires"
     pyarrow >=11.0,<11.1
     pymongo >=3.11,<5
-    pandas >=1.3.5,<2
+    pandas >=1.3.5,<3
 
 [options.package_data]
 pymongoarrow = *.pxd, *.pyx, *.pyi, *.so.*, *.dylib, *.dll, *.pyd

bindings/python/test/test_pandas.py

@@ -139,7 +139,7 @@ def _create_data(self):
         schema = {k: v.to_pandas_dtype() for k, v in arrow_schema.items()}
         schema["Int64"] = pd.Int64Dtype()
         schema["int"] = pd.Int32Dtype()
-        schema["str"] = "U8"
+        schema["str"] = "string"
         schema["datetime"] = "datetime64[ns]"
 
         data = pd.DataFrame(