Cover the repr dunders with tests

Author: rmnskb

python/pyarrow/tests/test_flight.py

@@ -47,13 +47,15 @@
         ServerAuthHandler, ClientAuthHandler,
         ServerMiddleware, ServerMiddlewareFactory,
         ClientMiddleware, ClientMiddlewareFactory,
+        FlightCallOptions,
     )
 except ImportError:
     flight = None
     FlightClient, FlightServerBase = object, object
     ServerAuthHandler, ClientAuthHandler = object, object
     ServerMiddleware, ServerMiddlewareFactory = object, object
     ClientMiddleware, ClientMiddlewareFactory = object, object
+    FlightCallOptions = object
 
 # Marks all of the tests in this module
 # Ignore these with pytest ... -m 'not flight'
@@ -2536,3 +2538,41 @@ def received_headers(self, headers):
         assert ("x-header-bin", b"header\x01value") in factory.headers
         assert ("x-trailer", "trailer-value") in factory.headers
         assert ("x-trailer-bin", b"trailer\x01value") in factory.headers
+
+
+@pytest.fixture
+def call_options_args(request):
+    match request.param:
+        case "default":
+            return {
+                "timeout": 3,
+                "headers": None,
+                "write_options": None,
+                "read_options": None,
+            }
+        case "all":
+            return {
+                "timeout": 7,
+                "headers": [(b"abc", b"def")],
+                "write_options": pa.ipc.IpcWriteOptions(compression="zstd"),
+                "read_options": pa.ipc.IpcReadOptions(use_threads=False),
+            }
+
+
+@pytest.mark.parametrize(
+    "call_options_args", ["default", "all"], indirect=True)
+def test_call_options_repr(call_options_args):
+    call_options = FlightCallOptions(**call_options_args)
+    repr = call_options.__repr__()
+
+    for arg, val in call_options_args.items():
+        if val is None:
+            continue
+
+        if arg not in ("read_options", "write_options"):
+            assert f"{arg}={val}" in repr
+            continue
+
+        val_repr = val.__repr__()
+        assert val_repr in repr
+

python/pyarrow/tests/test_ipc.py

@@ -1338,3 +1338,79 @@ def test_record_batch_file_writer_with_empty_metadata():
     buffer = sink.getvalue()
     with pa.ipc.open_file(buffer) as r:
         assert r.metadata is None
+
+
+def check_ipc_options_repr(options_obj, options_args):
+    options = options_obj(**options_args)
+    repr = options.__repr__()
+
+    for arg, val in options_args.items():
+        if val is None:
+            continue
+
+        value = val if not isinstance(val, str) else f"\"{val}\""
+
+        if arg == "ensure_alignment":
+            value = pa.ipc.Alignment(val).name
+        elif arg == "metadata_version":
+            value = pa.ipc.MetadataVersion(val).name
+
+        assert f"{arg}={value}" in repr
+
+
+@pytest.fixture
+def write_options_args(request):
+    match request.param:
+        case "default":
+            return {
+                "allow_64bit": False,
+                "use_legacy_format": False,
+                "metadata_version": pa.ipc.MetadataVersion.V5,
+                "compression": None,
+                "use_threads": True,
+                "emit_dictionary_deltas": False,
+                "unify_dictionaries": False,
+            }
+        case "all":
+            return {
+                "allow_64bit": True,
+                "use_legacy_format": True,
+                "metadata_version": pa.ipc.MetadataVersion.V4,
+                "compression": "zstd",
+                "use_threads": False,
+                "emit_dictionary_deltas": True,
+                "unify_dictionaries": True,
+            }
+
+
+@pytest.mark.parametrize(
+    "write_options_args", ["default", "all"], indirect=True)
+def test_write_options_repr(write_options_args):
+    # https://github.com/apache/arrow/issues/47358
+    check_ipc_options_repr(pa.ipc.IpcWriteOptions, write_options_args)
+
+
+@pytest.fixture
+def read_options_args(request):
+    match request.param:
+        case "default":
+            return {
+                "ensure_native_endian": True,
+                "ensure_alignment": pa.ipc.Alignment.Any,
+                "use_threads": True,
+                "included_fields": None,
+            }
+        case "all":
+            return {
+                "ensure_native_endian": False,
+                "ensure_alignment": pa.ipc.Alignment.DataTypeSpecific,
+                "use_threads": False,
+                "included_fields": [1, 2, 3],
+            }
+
+
+@pytest.mark.parametrize(
+    "read_options_args", ["default", "all"], indirect=True)
+def test_read_options_repr(read_options_args):
+    # https://github.com/apache/arrow/issues/47358
+    check_ipc_options_repr(pa.ipc.IpcReadOptions, read_options_args)