Skip to content

Commit

Permalink
missing markers (apache#969)
Browse files Browse the repository at this point in the history
Co-authored-by: Sung Yun <[email protected]>
  • Loading branch information
sungwy and sungwy authored Jul 26, 2024
1 parent 0213dab commit be5c426
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tests/integration/test_deletes.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,6 +105,7 @@ def test_partitioned_table_rewrite(spark: SparkSession, session_catalog: RestCat
assert tbl.scan().to_arrow().to_pydict() == {"number_partitioned": [11, 10], "number": [30, 30]}


@pytest.mark.integration
@pytest.mark.parametrize("format_version", [1, 2])
def test_rewrite_partitioned_table_with_null(spark: SparkSession, session_catalog: RestCatalog, format_version: int) -> None:
identifier = "default.table_partitioned_delete"
Expand Down Expand Up @@ -453,6 +454,7 @@ def test_delete_truncate(session_catalog: RestCatalog) -> None:
assert entries[0].status == ManifestEntryStatus.DELETED


@pytest.mark.integration
def test_delete_overwrite_table_with_null(session_catalog: RestCatalog) -> None:
arrow_schema = pa.schema([pa.field("ints", pa.int32())])
arrow_tbl = pa.Table.from_pylist(
Expand Down Expand Up @@ -492,6 +494,7 @@ def test_delete_overwrite_table_with_null(session_catalog: RestCatalog) -> None:
assert tbl.scan().to_arrow()["ints"].to_pylist() == [3, 4, 1, None]


@pytest.mark.integration
def test_delete_overwrite_table_with_nan(session_catalog: RestCatalog) -> None:
arrow_schema = pa.schema([pa.field("floats", pa.float32())])

Expand Down

0 comments on commit be5c426

Please sign in to comment.