diff --git a/tests/utils/test_manifest.py b/tests/utils/test_manifest.py index f2ae1e05ad..dfe62232a4 100644 --- a/tests/utils/test_manifest.py +++ b/tests/utils/test_manifest.py @@ -424,7 +424,7 @@ def test_write_manifest( assert manifest_entry.status == ManifestEntryStatus.ADDED assert manifest_entry.snapshot_id == 8744736658442914487 - assert manifest_entry.sequence_number == -1 if format_version == 1 else 3 + assert manifest_entry.sequence_number == (-1 if format_version == 1 else 3) assert isinstance(manifest_entry.data_file, DataFile) data_file = manifest_entry.data_file @@ -587,9 +587,9 @@ def test_write_manifest_list( assert manifest_file.manifest_length == 7989 assert manifest_file.partition_spec_id == 0 - assert manifest_file.content == ManifestContent.DATA if format_version == 1 else ManifestContent.DELETES - assert manifest_file.sequence_number == 0 if format_version == 1 else 3 - assert manifest_file.min_sequence_number == 0 if format_version == 1 else 3 + assert manifest_file.content == (ManifestContent.DATA if format_version == 1 else ManifestContent.DELETES) + assert manifest_file.sequence_number == (0 if format_version == 1 else 3) + assert manifest_file.min_sequence_number == (0 if format_version == 1 else 3) assert manifest_file.added_snapshot_id == 9182715666859759686 assert manifest_file.added_files_count == 3 assert manifest_file.existing_files_count == 0 @@ -616,8 +616,8 @@ def test_write_manifest_list( entry = entries[0] - assert entry.sequence_number == 0 if format_version == 1 else 3 - assert entry.file_sequence_number == 0 if format_version == 1 else 3 + assert entry.sequence_number == (0 if format_version == 1 else 3) + assert entry.file_sequence_number == (0 if format_version == 1 else 3) assert entry.snapshot_id == 8744736658442914487 assert entry.status == ManifestEntryStatus.ADDED