Skip to content

Commit a6b18a2

Browse files
committed
fix: ensure correct status assertion for failed pages in tests
Signed-off-by: jhchoi1182 <jhchoi1182@gmail.com>
1 parent 7772e97 commit a6b18a2

1 file changed

Lines changed: 6 additions & 8 deletions

File tree

tests/test_failed_pages.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -97,10 +97,9 @@ def test_failed_pages_added_to_document_1page(skipped_1page_path):
9797
result = converter.convert(skipped_1page_path, raises_on_error=False)
9898

9999
# Document should have partial success due to failed page(s)
100-
assert result.status in [
101-
ConversionStatus.SUCCESS,
102-
ConversionStatus.PARTIAL_SUCCESS,
103-
], f"Unexpected status: {result.status}"
100+
assert result.status == ConversionStatus.PARTIAL_SUCCESS, (
101+
f"Unexpected status: {result.status}"
102+
)
104103

105104
# Get expected page count from input
106105
expected_page_count = result.input.page_count
@@ -141,10 +140,9 @@ def test_failed_pages_added_to_document_2pages(skipped_2pages_path):
141140
result = converter.convert(skipped_2pages_path, raises_on_error=False)
142141

143142
# Document should have partial success due to failed page(s)
144-
assert result.status in [
145-
ConversionStatus.SUCCESS,
146-
ConversionStatus.PARTIAL_SUCCESS,
147-
], f"Unexpected status: {result.status}"
143+
assert result.status == ConversionStatus.PARTIAL_SUCCESS, (
144+
f"Unexpected status: {result.status}"
145+
)
148146

149147
# Get expected page count from input
150148
expected_page_count = result.input.page_count

0 commit comments

Comments
 (0)