//*****************************************************************************// Copyright 2017-2019 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.//*****************************************************************************#include "gtest/gtest.h"#include "ngraph/ngraph.hpp"#include "util/type_prop.hpp"usingnamespacestd;usingnamespacengraph;TEST(type_prop,squeeze){autoparam=make_shared<op::Parameter>(element::f32,Shape{1,4,1,4,1,8});autoaxes_node=make_shared<ngraph::op::Constant>(element::u64,Shape{2},vector<int64_t>{0,2});autosqueeze=make_shared<op::Squeeze>(param,axes_node);ASSERT_EQ(squeeze->get_element_type(),element::f32);ASSERT_EQ(squeeze->get_shape(),(Shape{4,4,1,8}));axes_node=make_shared<ngraph::op::Constant>(element::u64,Shape{0},vector<int64_t>{});autosqueeze_default_axes=make_shared<op::Squeeze>(param,axes_node);ASSERT_EQ(squeeze_default_axes->get_element_type(),element::f32);ASSERT_EQ(squeeze_default_axes->get_shape(),(Shape{4,4,8}));}