@@ -952,33 +952,6 @@ def test_adjust_contrast():
952
952
torch .testing .assert_close (y_np , y_ans )
953
953
954
954
955
- @pytest .mark .skipif (Image .__version__ >= "7" , reason = "Temporarily disabled" )
956
- def test_adjust_saturation ():
957
- x_shape = [2 , 2 , 3 ]
958
- x_data = [0 , 5 , 13 , 54 , 135 , 226 , 37 , 8 , 234 , 90 , 255 , 1 ]
959
- x_np = np .array (x_data , dtype = np .uint8 ).reshape (x_shape )
960
- x_pil = Image .fromarray (x_np , mode = "RGB" )
961
-
962
- # test 0
963
- y_pil = F .adjust_saturation (x_pil , 1 )
964
- y_np = np .array (y_pil )
965
- torch .testing .assert_close (y_np , x_np )
966
-
967
- # test 1
968
- y_pil = F .adjust_saturation (x_pil , 0.5 )
969
- y_np = np .array (y_pil )
970
- y_ans = [2 , 4 , 8 , 87 , 128 , 173 , 39 , 25 , 138 , 133 , 215 , 88 ]
971
- y_ans = np .array (y_ans , dtype = np .uint8 ).reshape (x_shape )
972
- torch .testing .assert_close (y_np , y_ans )
973
-
974
- # test 2
975
- y_pil = F .adjust_saturation (x_pil , 2 )
976
- y_np = np .array (y_pil )
977
- y_ans = [0 , 6 , 22 , 0 , 149 , 255 , 32 , 0 , 255 , 4 , 255 , 0 ]
978
- y_ans = np .array (y_ans , dtype = np .uint8 ).reshape (x_shape )
979
- torch .testing .assert_close (y_np , y_ans )
980
-
981
-
982
955
def test_adjust_hue ():
983
956
x_shape = [2 , 2 , 3 ]
984
957
x_data = [0 , 5 , 13 , 54 , 135 , 226 , 37 , 8 , 234 , 90 , 255 , 1 ]
0 commit comments