Skip to content

Commit 06c4a4d

Browse files
committed
fix test
1 parent 50b7b39 commit 06c4a4d

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/integration/test_reads.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -651,10 +651,10 @@ def test_scan_branch(catalog: Catalog, format_version: int) -> None:
651651
def test_filter_on_new_column(catalog: Catalog) -> None:
652652
test_table_add_column = catalog.load_table("default.test_table_add_column")
653653
arrow_table = test_table_add_column.scan(row_filter="b == '2'").to_arrow()
654-
assert arrow_table["b"].to_pylist() == ["2"]
654+
assert arrow_table["b"].to_pylist() == ["2", None]
655655

656656
arrow_table = test_table_add_column.scan(row_filter="b is not null").to_arrow()
657-
assert arrow_table["b"].to_pylist() == ["2"]
657+
assert arrow_table["b"].to_pylist() == ["2", None]
658658

659659
arrow_table = test_table_add_column.scan(row_filter="b is null").to_arrow()
660660
assert arrow_table["b"].to_pylist() == [None]
@@ -668,7 +668,7 @@ def test_filter_case_sensitive_by_default(catalog: Catalog) -> None:
668668
assert "2" in arrow_table["b"].to_pylist()
669669

670670
arrow_table = test_table_add_column.scan(row_filter="b == '2'").to_arrow()
671-
assert arrow_table["b"].to_pylist() == ["2"]
671+
assert arrow_table["b"].to_pylist() == ["2", None]
672672

673673
with pytest.raises(ValueError) as e:
674674
_ = test_table_add_column.scan(row_filter="B == '2'").to_arrow()
@@ -683,7 +683,7 @@ def test_filter_case_sensitive(catalog: Catalog) -> None:
683683
assert "2" in arrow_table["b"].to_pylist()
684684

685685
arrow_table = test_table_add_column.scan(row_filter="b == '2'", case_sensitive=True).to_arrow()
686-
assert arrow_table["b"].to_pylist() == ["2"]
686+
assert arrow_table["b"].to_pylist() == ["2", None]
687687

688688
with pytest.raises(ValueError) as e:
689689
_ = test_table_add_column.scan(row_filter="B == '2'", case_sensitive=True).to_arrow()
@@ -698,10 +698,10 @@ def test_filter_case_insensitive(catalog: Catalog) -> None:
698698
assert "2" in arrow_table["b"].to_pylist()
699699

700700
arrow_table = test_table_add_column.scan(row_filter="b == '2'", case_sensitive=False).to_arrow()
701-
assert arrow_table["b"].to_pylist() == ["2"]
701+
assert arrow_table["b"].to_pylist() == ["2", None]
702702

703703
arrow_table = test_table_add_column.scan(row_filter="B == '2'", case_sensitive=False).to_arrow()
704-
assert arrow_table["b"].to_pylist() == ["2"]
704+
assert arrow_table["b"].to_pylist() == ["2", None]
705705

706706

707707
@pytest.mark.integration

0 commit comments

Comments
 (0)