From 546b8527bcd23018d4c1041b7234c98c0e7c77e1 Mon Sep 17 00:00:00 2001 From: JingxianKe <983231802@qq.com> Date: Thu, 3 Aug 2023 10:51:58 +0800 Subject: [PATCH 1/2] Update test_runtime_ort.py --- test/test_runtime_ort.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_runtime_ort.py b/test/test_runtime_ort.py index 483bde48..1be9e3ed 100644 --- a/test/test_runtime_ort.py +++ b/test/test_runtime_ort.py @@ -132,5 +132,5 @@ def test_onnx_export_misbatch(self, arch): self.run_model(model, [[img_one, img_two], [img_two, img_one, img_dummy]]) # Test exported model on images of misbatch - with pytest.raises(ValueError, match="Model requires 3 inputs. Input Feed contains 2"): + with pytest.raises(ValueError, match="Required inputs (['image3']) are missing from input feed"): self.run_model(model, [[img_two, img_one, img_dummy], [img_one, img_two]]) From 6631b6ebef67c76abd4cf9fc57324e91b9dc83cf Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Thu, 3 Aug 2023 13:43:40 +0800 Subject: [PATCH 2/2] Update test_runtime_ort.py --- test/test_runtime_ort.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/test_runtime_ort.py b/test/test_runtime_ort.py index 1be9e3ed..7ca02f06 100644 --- a/test/test_runtime_ort.py +++ b/test/test_runtime_ort.py @@ -132,5 +132,8 @@ def test_onnx_export_misbatch(self, arch): self.run_model(model, [[img_one, img_two], [img_two, img_one, img_dummy]]) # Test exported model on images of misbatch - with pytest.raises(ValueError, match="Required inputs (['image3']) are missing from input feed"): + with pytest.raises( + ValueError, + match="Required inputs (['image3']) are missing from input feed (['image1', 'image2']).", + ): self.run_model(model, [[img_two, img_one, img_dummy], [img_one, img_two]])