adbc-driver-postgresql 1.0.0__tar.gz → 1.2.0__tar.gz

{adbc_driver_postgresql-1.0.0/adbc_driver_postgresql.egg-info → adbc_driver_postgresql-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: adbc-driver-postgresql
-Version: 1.0.0
+Version: 1.2.0
 Summary: A libpq-based ADBC driver for working with PostgreSQL.
 Author-email: Apache Arrow Developers <dev@arrow.apache.org>
 License: Apache-2.0

{adbc_driver_postgresql-1.0.0 → adbc_driver_postgresql-1.2.0}/adbc_driver_postgresql/_static_version.py

@@ -20,9 +20,9 @@
 
 # This file is part of 'miniver': https://github.com/jbweston/miniver
 
-version = "1.0.0"
+version = "1.2.0"
 
 # These values are only set if the distribution was created with 'git archive'
 # NOTE: must add an export-subst to .gitattributes!
-refnames = "HEAD, tag: apache-arrow-adbc-12-rc4, origin/maint-12"
-git_hash = "50cb9de62"
+refnames = "HEAD, tag: apache-arrow-adbc-14-rc0, origin/maint-14"
+git_hash = "15f330ce9"

{adbc_driver_postgresql-1.0.0 → adbc_driver_postgresql-1.2.0/adbc_driver_postgresql.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: adbc-driver-postgresql
-Version: 1.0.0
+Version: 1.2.0
 Summary: A libpq-based ADBC driver for working with PostgreSQL.
 Author-email: Apache Arrow Developers <dev@arrow.apache.org>
 License: Apache-2.0

{adbc_driver_postgresql-1.0.0 → adbc_driver_postgresql-1.2.0}/tests/test_dbapi.py

@@ -15,9 +15,11 @@
 # specific language governing permissions and limitations
 # under the License.
 
+import string
 from pathlib import Path
 from typing import Generator
 
+import numpy
 import pyarrow
 import pyarrow.dataset
 import pytest
@@ -146,7 +148,7 @@ def test_query_execute_schema(postgres: dbapi.Connection) -> None:
 def test_query_invalid(postgres: dbapi.Connection) -> None:
     with postgres.cursor() as cur:
         with pytest.raises(
-            postgres.ProgrammingError, match="failed to prepare query"
+            postgres.ProgrammingError, match="Failed to prepare query"
         ) as excinfo:
             cur.execute("SELECT * FROM tabledoesnotexist")
 
@@ -410,3 +412,15 @@ def test_ingest_temporary(postgres: dbapi.Connection) -> None:
         assert cur.fetch_arrow_table() == temp2
         cur.execute("SELECT * FROM temporary")
         assert cur.fetch_arrow_table() == temp2
+
+
+def test_ingest_large(postgres: dbapi.Connection) -> None:
+    """Regression test for #1921."""
+    # More than 1 GiB of data in one batch
+    arr = pyarrow.array(numpy.random.randint(-100, 100, size=4_000_000))
+    batch = pyarrow.RecordBatch.from_pydict(
+        {char: arr for char in string.ascii_lowercase}
+    )
+    table = pyarrow.Table.from_batches([batch] * 4)
+    with postgres.cursor() as cur:
+        cur.adbc_ingest("test_ingest_large", table, mode="replace")