From cbad43f3a55f923510b7611a7bea842c52f20881 Mon Sep 17 00:00:00 2001 From: Anastasia Kuporosova Date: Fri, 29 May 2020 21:20:16 +0300 Subject: [PATCH] [Python API] Fix PreProcessInfo tests (#690) --- .../ie_bridges/python/tests/test_PreProcessInfo.py | 32 ++++++++++++++-------- 1 file changed, 20 insertions(+), 12 deletions(-) diff --git a/inference-engine/ie_bridges/python/tests/test_PreProcessInfo.py b/inference-engine/ie_bridges/python/tests/test_PreProcessInfo.py index 499fdb3..5cfc996 100644 --- a/inference-engine/ie_bridges/python/tests/test_PreProcessInfo.py +++ b/inference-engine/ie_bridges/python/tests/test_PreProcessInfo.py @@ -8,45 +8,53 @@ from conftest import model_path test_net_xml, test_net_bin = model_path() -def get_preprocess_info(): +def test_preprocess_info(): ie_core = IECore() net = ie_core.read_network(model=test_net_xml, weights=test_net_bin) - return net.input_info["data"].preprocess_info - - -def test_preprocess_info(): - assert isinstance(get_preprocess_info(), PreProcessInfo) + assert isinstance(net.input_info["data"].preprocess_info, PreProcessInfo) def test_color_format(): - preprocess_info = get_preprocess_info() + ie_core = IECore() + net = ie_core.read_network(model=test_net_xml, weights=test_net_bin) + preprocess_info = net.input_info["data"].preprocess_info assert preprocess_info.color_format == ColorFormat.RAW def test_color_format_setter(): - preprocess_info = get_preprocess_info() + ie_core = IECore() + net = ie_core.read_network(model=test_net_xml, weights=test_net_bin) + preprocess_info = net.input_info["data"].preprocess_info preprocess_info.color_format = ColorFormat.BGR assert preprocess_info.color_format == ColorFormat.BGR def test_resize_algorithm(): - preprocess_info = get_preprocess_info() + ie_core = IECore() + net = ie_core.read_network(model=test_net_xml, weights=test_net_bin) + preprocess_info = net.input_info["data"].preprocess_info assert preprocess_info.resize_algorithm == ResizeAlgorithm.NO_RESIZE def test_resize_algorithm_setter(): - preprocess_info = get_preprocess_info() + ie_core = IECore() + net = ie_core.read_network(model=test_net_xml, weights=test_net_bin) + preprocess_info = net.input_info["data"].preprocess_info preprocess_info.resize_algorithm = ResizeAlgorithm.RESIZE_BILINEAR assert preprocess_info.resize_algorithm == ResizeAlgorithm.RESIZE_BILINEAR def test_mean_variant(): - preprocess_info = get_preprocess_info() + ie_core = IECore() + net = ie_core.read_network(model=test_net_xml, weights=test_net_bin) + preprocess_info = net.input_info["data"].preprocess_info assert preprocess_info.mean_variant == MeanVariant.NONE def test_mean_variant_setter(): - preprocess_info = get_preprocess_info() + ie_core = IECore() + net = ie_core.read_network(model=test_net_xml, weights=test_net_bin) + preprocess_info = net.input_info["data"].preprocess_info preprocess_info.mean_variant = MeanVariant.MEAN_IMAGE assert preprocess_info.mean_variant == MeanVariant.MEAN_IMAGE -- 2.7.4