Update test script

tests/test_image_reader.py

@@ -5,6 +5,8 @@ import numpy as np
 import pytest
 from PIL import Image
 
+INPUT_HEIGHT, INPUT_WIDTH = 600, 500
+
 
 @pytest.mark.parametrize('dtype,extension,dim', [
     (np.uint8, '.png', 1),
@@ -31,7 +33,9 @@ def test_image_reader(dtype, extension, dim):
     image_r = np.array(Image.open(tf.name))
     image_r = cv2.normalize(
         image_r, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U)
+    image_r = cv2.resize(
+        image_r, (INPUT_WIDTH, INPUT_HEIGHT), interpolation=cv2.INTER_LINEAR)
 
     assert image_r.dtype == np.uint8
-    assert image_r.shape[:2] == (225, 225)
+    assert image_r.shape[:2] == (INPUT_HEIGHT, INPUT_WIDTH)
     assert np.min(image_r) >= 0 and np.max(image_r) <= 255