Skip to content

Commit 38e61ea

Browse files
author
Abanoub Doss
committed
test: update arrow input error expectations
1 parent 286a7de commit 38e61ea

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

tests/integration/test_writes/test_partitioned_writes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -768,7 +768,7 @@ def test_invalid_arguments(spark: SparkSession, session_catalog: Catalog) -> Non
768768
properties={"format-version": "1"},
769769
)
770770

771-
with pytest.raises(ValueError, match="Expected pa.Table or pa.RecordBatchReader, got: not a df"):
771+
with pytest.raises(ValueError, match="Expected pa.Table, pa.RecordBatchReader, or an object implementing"):
772772
tbl.append("not a df")
773773

774774

tests/integration/test_writes/test_writes.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -791,10 +791,10 @@ def test_invalid_arguments(spark: SparkSession, session_catalog: Catalog, arrow_
791791
identifier = "default.arrow_data_files"
792792
tbl = _create_table(session_catalog, identifier, {"format-version": "1"}, [])
793793

794-
with pytest.raises(ValueError, match="Expected pa.Table or pa.RecordBatchReader, got: not a df"):
794+
with pytest.raises(ValueError, match="Expected pa.Table, pa.RecordBatchReader, or an object implementing"):
795795
tbl.overwrite("not a df")
796796

797-
with pytest.raises(ValueError, match="Expected pa.Table or pa.RecordBatchReader, got: not a df"):
797+
with pytest.raises(ValueError, match="Expected pa.Table, pa.RecordBatchReader, or an object implementing"):
798798
tbl.append("not a df")
799799

800800

0 commit comments

Comments
 (0)