Skip to content

Commit aebcb48

Browse files
committed
improve test
1 parent 93de319 commit aebcb48

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

tests/table/test_upsert.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -528,6 +528,7 @@ def test_upsert_with_nulls(catalog: Catalog) -> None:
528528
data_with_null = pa.Table.from_pylist(
529529
[
530530
{"foo": "apple", "bar": None, "baz": False},
531+
{"foo": "banana", "bar": None, "baz": False},
531532
],
532533
schema=schema,
533534
)
@@ -544,4 +545,10 @@ def test_upsert_with_nulls(catalog: Catalog) -> None:
544545
upd = table.upsert(data_without_null, join_cols=["foo"])
545546
assert upd.rows_updated == 1
546547
assert upd.rows_inserted == 0
547-
assert table.scan().to_arrow() == data_without_null
548+
assert table.scan().to_arrow() == pa.Table.from_pylist(
549+
[
550+
{"foo": "apple", "bar": 7, "baz": False},
551+
{"foo": "banana", "bar": None, "baz": False},
552+
],
553+
schema=schema,
554+
)

0 commit comments

Comments
 (0)