diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2018-09-18 16:53:40 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2018-09-18 16:53:40 +0900 |
commit | 91f4ba45449f700a047a4aeea00b1a7c84e94c75 (patch) | |
tree | c60eecdba0861c51010fb0519f8a59668d90a6d2 /runtimes/neurun/test/graph/operand/Set.cc | |
parent | 07659ccd9fe7b1cf1547cc6cad78bcf489f0a361 (diff) | |
download | nnfw-7710781997e1cb5e9499c7aec50c9af039473c00.tar.gz nnfw-7710781997e1cb5e9499c7aec50c9af039473c00.tar.bz2 nnfw-7710781997e1cb5e9499c7aec50c9af039473c00.zip |
Imported Upstream version 0.2upstream/0.2submit/tizen/20180918.075952
Diffstat (limited to 'runtimes/neurun/test/graph/operand/Set.cc')
-rw-r--r-- | runtimes/neurun/test/graph/operand/Set.cc | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/runtimes/neurun/test/graph/operand/Set.cc b/runtimes/neurun/test/graph/operand/Set.cc new file mode 100644 index 000000000..176cedc38 --- /dev/null +++ b/runtimes/neurun/test/graph/operand/Set.cc @@ -0,0 +1,48 @@ +/* + * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved + * + * 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 "graph/operand/Set.h" + +TEST(graph_operand_Set, set_test) +{ + neurun::graph::operand::Set set; + + ::neurun::graph::operand::Shape shape0{3}; + shape0.dim(0) = 1; + shape0.dim(1) = 2; + shape0.dim(2) = 3; + + ::neurun::graph::operand::Shape shape1{4}; + shape1.dim(0) = 10; + shape1.dim(1) = 20; + shape1.dim(2) = 30; + shape1.dim(3) = 40; + + ::neurun::graph::operand::TypeInfo type{ANEURALNETWORKS_TENSOR_INT32, 0, 0}; + + set.append(shape0, type); + set.append(shape1, type); + + ASSERT_EQ(set.exist(neurun::graph::operand::Index{0u}), true); + ASSERT_EQ(set.exist(neurun::graph::operand::Index{1u}), true); + ASSERT_EQ(set.exist(neurun::graph::operand::Index{2u}), false); + + ASSERT_EQ(set.at(neurun::graph::operand::Index{0u}).shape().dim(0), 1); + ASSERT_EQ(set.at(neurun::graph::operand::Index{0u}).shape().dim(1), 2); + ASSERT_EQ(set.at(neurun::graph::operand::Index{0u}).shape().dim(2), 3); +} |