054a7cdf8d
* Enable ngraph python tests * Refactor and unify ngraph with onnx python tests * Revert deprecated test cases * Set ngraph and onnx python tests as a one test suite execution * Change unstrict Xfails to strict ones * Update after review: - add model zoo to onnx tests, - improvements of tests * Revert mounting zoo models dir Co-authored-by: Michał Karzyński <4430709+postrational@users.noreply.github.com>
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
# ******************************************************************************
|
|
# Copyright 2017-2020 Intel Corporation
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ******************************************************************************
|
|
import numpy as np
|
|
|
|
import ngraph as ng
|
|
from tests.runtime import get_runtime
|
|
from tests import xfail_issue_34323
|
|
|
|
|
|
@xfail_issue_34323
|
|
def test_split():
|
|
runtime = get_runtime()
|
|
input_tensor = ng.constant(np.array([0, 1, 2, 3, 4, 5], dtype=np.int32))
|
|
axis = ng.constant(0, dtype=np.int64)
|
|
splits = 3
|
|
|
|
split_node = ng.split(input_tensor, axis, splits)
|
|
computation = runtime.computation(split_node)
|
|
split_results = computation()
|
|
expected_results = np.array([[0, 1], [2, 3], [4, 5]], dtype=np.int32)
|
|
assert np.allclose(split_results, expected_results)
|
|
|
|
|
|
@xfail_issue_34323
|
|
def test_variadic_split():
|
|
runtime = get_runtime()
|
|
input_tensor = ng.constant(np.array([[0, 1, 2, 3, 4, 5], [6, 7, 8, 9, 10, 11]], dtype=np.int32))
|
|
axis = ng.constant(1, dtype=np.int64)
|
|
splits = ng.constant(np.array([2, 4], dtype=np.int64))
|
|
|
|
v_split_node = ng.variadic_split(input_tensor, axis, splits)
|
|
computation = runtime.computation(v_split_node)
|
|
results = computation()
|
|
split0 = np.array([[0, 1], [6, 7]], dtype=np.int32)
|
|
split1 = np.array([[2, 3, 4, 5], [8, 9, 10, 11]], dtype=np.int32)
|
|
|
|
assert np.allclose(results[0], split0)
|
|
assert np.allclose(results[1], split1)
|