Skip to content

Commit 3489c5f

Browse files
Import ImageReader in test_reader.py
1 parent 1e3ae8f commit 3489c5f

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

tests/test_reader.py

+1-10
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from PIL import Image
55
import numpy as np
66
import io
7-
from facetorch.analyzer.reader import UniversalReader, TensorReader
7+
from facetorch.analyzer.reader import UniversalReader, TensorReader, ImageReader
88

99

1010
@pytest.mark.integration
@@ -81,15 +81,6 @@ def test_read_image_from_path(cfg, analyzer):
8181
assert result.tensor is not None
8282

8383

84-
def test_read_image_from_path(cfg, analyzer):
85-
if not isinstance(analyzer.reader, UniversalReader):
86-
pytest.skip("Only UniversalReader is used for this test.")
87-
result = analyzer.reader.run(cfg.path_image)
88-
assert isinstance(result, facetorch.datastruct.ImageData)
89-
assert result.img is not None
90-
assert result.tensor is not None
91-
92-
9384
def test_read_tensor(analyzer):
9485
if not isinstance(analyzer.reader, UniversalReader) or not isinstance(
9586
analyzer.reader, TensorReader

0 commit comments

Comments
 (0)