|
|
|
@@ -86,12 +86,8 @@ def test_five_crop_error_msg(): |
|
|
|
transform = vision.ComposeOp(transforms) |
|
|
|
data = data.map(input_columns=["image"], operations=transform()) |
|
|
|
|
|
|
|
with pytest.raises(RuntimeError) as info: |
|
|
|
with pytest.raises(RuntimeError): |
|
|
|
data.create_tuple_iterator().__next__() |
|
|
|
error_msg = "TypeError: img should be PIL Image or Numpy array. Got <class 'tuple'>" |
|
|
|
|
|
|
|
# error msg comes from ToTensor() |
|
|
|
assert error_msg in str(info.value) |
|
|
|
|
|
|
|
|
|
|
|
def test_five_crop_md5(): |
|
|
|
|