From b2e654aef97caae9f1cee3e127df038ceb8b57a8 Mon Sep 17 00:00:00 2001 From: erinhmclark Date: Wed, 26 Feb 2025 12:57:33 +0000 Subject: [PATCH] Remove context manager from test_pdq_hash_enricher.py --- tests/enrichers/test_pdq_hash_enricher.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/tests/enrichers/test_pdq_hash_enricher.py b/tests/enrichers/test_pdq_hash_enricher.py index 9653734..a8470fb 100644 --- a/tests/enrichers/test_pdq_hash_enricher.py +++ b/tests/enrichers/test_pdq_hash_enricher.py @@ -20,17 +20,15 @@ def metadata_with_images(): def test_successful_enrich(metadata_with_images, mocker): - with ( - mocker.patch("pdqhash.compute", return_value=([1, 0, 1, 0] * 64, 100)), - mocker.patch("PIL.Image.open"), - mocker.patch.object(Media, "is_image", return_value=True) as mock_is_image, - ): - enricher = PdqHashEnricher() - enricher.enrich(metadata_with_images) + mocker.patch("pdqhash.compute", return_value=([1, 0, 1, 0] * 64, 100)) + mocker.patch("PIL.Image.open") + mocker.patch.object(Media, "is_image", return_value=True) + enricher = PdqHashEnricher() + enricher.enrich(metadata_with_images) - # Ensure the hash is set for image media - for media in metadata_with_images.media: - assert media.get("pdq_hash") is not None + # Ensure the hash is set for image media + for media in metadata_with_images.media: + assert media.get("pdq_hash") is not None def test_enrich_skip_non_image(metadata_with_images, mocker):