Danila-Pechenev commited on
Commit
c63c2cd
1 Parent(s): 28de88c
Files changed (1) hide show
  1. test/test_model.py +4 -4
test/test_model.py CHANGED
@@ -7,7 +7,7 @@ from ..app import model
7
  model_instance: keras.Model = model.create_model()
8
 
9
 
10
- def template(filename: str) -> (int, int, Image.Image):
11
  image: Image.Image = Image.open(filename)
12
  width: int
13
  height: int
@@ -22,7 +22,7 @@ def test_image_jpg():
22
  width: int
23
  height: int
24
  output_image: Image.Image
25
- width, height, output_image = template(os.path.join(os.getcwd(), "test_images/test1.jpg"))
26
 
27
  assert width == output_image.size[0]
28
  assert height == output_image.size[1]
@@ -32,7 +32,7 @@ def test_image_png():
32
  width: int
33
  height: int
34
  output_image: Image.Image
35
- width, height, output_image = template(os.path.join(os.getcwd(), "test_images/test2.png"))
36
 
37
  assert width == output_image.size[0]
38
  assert height == output_image.size[1]
@@ -42,7 +42,7 @@ def test_image_jpeg():
42
  width: int
43
  height: int
44
  output_image: Image.Image
45
- width, height, output_image = template(os.path.join(os.getcwd(), "test_images/test3.jpeg"))
46
 
47
  assert width == output_image.size[0]
48
  assert height == output_image.size[1]
 
7
  model_instance: keras.Model = model.create_model()
8
 
9
 
10
+ def template(filename: str) -> tuple[int, int, Image.Image]:
11
  image: Image.Image = Image.open(filename)
12
  width: int
13
  height: int
 
22
  width: int
23
  height: int
24
  output_image: Image.Image
25
+ width, height, output_image = template(os.path.join(os.getcwd(), "test/test_images/test1.jpg"))
26
 
27
  assert width == output_image.size[0]
28
  assert height == output_image.size[1]
 
32
  width: int
33
  height: int
34
  output_image: Image.Image
35
+ width, height, output_image = template(os.path.join(os.getcwd(), "test/test_images/test2.png"))
36
 
37
  assert width == output_image.size[0]
38
  assert height == output_image.size[1]
 
42
  width: int
43
  height: int
44
  output_image: Image.Image
45
+ width, height, output_image = template(os.path.join(os.getcwd(), "test/test_images/test3.jpeg"))
46
 
47
  assert width == output_image.size[0]
48
  assert height == output_image.size[1]