updates mock

Author: chalmerlowe

tests/unit/test__pandas_helpers.py

@@ -2262,6 +2262,7 @@ def test_download_arrow_row_iterator_timeout(module_under_test):
     # Mock page with to_arrow method
     mock_page = mock.Mock()
     mock_page.to_arrow.return_value = pyarrow.RecordBatch.from_pydict({"name": ["foo"]})
+    mock_page.__iter__ = lambda self: iter(["row1"])
 
     def slow_pages():
         # First page yields quickly