From 2023581893fdf145ddc8778f799b33735ba456f5 Mon Sep 17 00:00:00 2001 From: Agam Sanghera Date: Sat, 18 Jan 2025 11:11:48 -0800 Subject: [PATCH] Updated tests --- tests/test_augment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_augment.py b/tests/test_augment.py index 0068d28..33bd98b 100644 --- a/tests/test_augment.py +++ b/tests/test_augment.py @@ -38,10 +38,10 @@ def test_augment(): # call individual tests assert np.isclose(blur(input),output_image,atol=.01).all, "Incorrect output, function not working as expected" # Check if invalid functions raise a ValueError - invalid_operations = [(blur,1)(flip, 1), (scale, 0.5), (grayscale, 0.7), (lambda x: x, 1)] # Invalid function (lambda) + invalid_operations = [(blur,1),(flip, 1), (scale, 0.5), (grayscale, 0.7), (lambda x: x, 1)] # Invalid function (lambda) with pytest.raises(ValueError, match="Function is not allowed"): augment(input_image, invalid_operations) - invalid_operations_2 = [(flip, 1), (scale, 0.5), (blur, 3), (lambda x: x, 1)] + invalid_operations_2 = [(flip, 1), (scale, 0.5), (blur, 1), (lambda x: x, 1)] with pytest.raises(ValueError): augment(input_image, invalid_operations_2)