Add ViTImageProcessorFast to tests (#31424)

* Add ViTImageProcessor to tests

* Correct data format

* Review comments

tests/models/swin2sr/test_image_processing_swin2sr.py

@@ -72,6 +72,8 @@ class Swin2SRImageProcessingTester(unittest.TestCase):
 
         if isinstance(img, Image.Image):
             input_width, input_height = img.size
+        elif isinstance(img, np.ndarray):
+            input_height, input_width = img.shape[-3:-1]
         else:
             input_height, input_width = img.shape[-2:]
 
@@ -160,7 +162,7 @@ class Swin2SRImageProcessingTest(ImageProcessingTestMixin, unittest.TestCase):
 
         # Test not batched input
         encoded_images = image_processing(
-            image_inputs[0], return_tensors="pt", input_data_format="channels_first"
+            image_inputs[0], return_tensors="pt", input_data_format="channels_last"
         ).pixel_values
         expected_output_image_shape = self.image_processor_tester.expected_output_image_shape([image_inputs[0]])
         self.assertEqual(tuple(encoded_images.shape), (1, *expected_output_image_shape))