test: add garbage collection test for arrow stream reader in DataFrame

Author: kosiew

python/tests/test_dataframe.py

@@ -1929,6 +1929,28 @@ def test_iter_then_arrow_c_stream_gc():
         gc.collect()
 
 
+def test_iter_arrow_stream_reader_gc():
+    """Iterate over DataFrame, consume __arrow_c_stream__ reader, then gc."""
+
+    ctx = SessionContext()
+    df = ctx.from_pydict({"a": [1, 2]})
+
+    batches = list(df)
+    assert len(batches) == 1
+    assert batches[0].column(0).to_pylist() == [1, 2]
+
+    reader = pa.RecordBatchReader._import_from_c_capsule(df.__arrow_c_stream__())
+    table = reader.read_all()
+    assert table.column(0).to_pylist() == [1, 2]
+
+    del table
+    del reader
+    del batches
+    del df
+    del ctx
+    gc.collect()
+
+
 def test_to_pylist(df):
     # Convert datafusion dataframe to Python list
     pylist = df.to_pylist()