Skip to content

Commit

Permalink
Remove image type assertions but keep all other assertions in
Browse files Browse the repository at this point in the history
  • Loading branch information
niwilso committed Jul 11, 2023
1 parent ca230c3 commit 049f4ad
Showing 1 changed file with 10 additions and 10 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -61,12 +61,12 @@ def test_verify_correctly(
)
test_all_labels = set(expected_ocr_results_labels).union(set(test_ocr_results_labels))

# # Assert
# Assert
# assert type(test_image_verify) == PIL.Image.Image
# assert len(test_common_labels) / len(test_all_labels) >= 0.5
# _strip_score(expected_analyzer_results)
# _strip_score(test_analyzer_results_formatted)
# assert _set_of_tuples(test_analyzer_results_formatted) == _set_of_tuples(expected_analyzer_results)
assert len(test_common_labels) / len(test_all_labels) >= 0.5
_strip_score(expected_analyzer_results)
_strip_score(test_analyzer_results_formatted)
assert _set_of_tuples(test_analyzer_results_formatted) == _set_of_tuples(expected_analyzer_results)

def test_eval_dicom_correctly(
mock_instance: pydicom.dataset.FileDataset,
Expand Down Expand Up @@ -100,11 +100,11 @@ def test_eval_dicom_correctly(
ocr_kwargs = None
)

# # Assert
assert type(test_image_eval) == PIL.Image.Image
# _strip_score(test_eval_results['all_positives'])
# _strip_score(expected_results['all_positives'])
# assert test_eval_results == expected_results
# Assert
# assert type(test_image_eval) == PIL.Image.Image
_strip_score(test_eval_results['all_positives'])
_strip_score(expected_results['all_positives'])
assert test_eval_results == expected_results

def _strip_score(analyzer_results_to_remove_score_from):
[result.pop('score') for result in analyzer_results_to_remove_score_from]
Expand Down

0 comments on commit 049f4ad

Please sign in to comment.