2 * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include "luci/IR/Nodes/CircleGatherNd.h"
19 #include "luci/IR/CircleDialect.h"
20 #include "luci/IR/CircleNodeVisitor.h"
22 #include <gtest/gtest.h>
24 TEST(CircleGatherNdTest, constructor)
26 luci::CircleGatherNd gather_nd_node;
28 ASSERT_EQ(luci::CircleDialect::get(), gather_nd_node.dialect());
29 ASSERT_EQ(luci::CircleOpcode::GATHER_ND, gather_nd_node.opcode());
31 ASSERT_EQ(nullptr, gather_nd_node.params());
32 ASSERT_EQ(nullptr, gather_nd_node.indices());
35 TEST(CircleGatherNdTest, input_NEG)
37 luci::CircleGatherNd gather_nd_node;
38 luci::CircleGatherNd node;
40 gather_nd_node.params(&node);
41 gather_nd_node.indices(&node);
42 ASSERT_NE(nullptr, gather_nd_node.params());
43 ASSERT_NE(nullptr, gather_nd_node.indices());
45 gather_nd_node.params(nullptr);
46 gather_nd_node.indices(nullptr);
47 ASSERT_EQ(nullptr, gather_nd_node.params());
48 ASSERT_EQ(nullptr, gather_nd_node.indices());
51 TEST(CircleGatherNdTest, arity_NEG)
53 luci::CircleGatherNd gather_nd_node;
55 ASSERT_NO_THROW(gather_nd_node.arg(1));
56 ASSERT_THROW(gather_nd_node.arg(2), std::out_of_range);
59 TEST(CircleGatherNdTest, visit_mutable_NEG)
61 struct TestVisitor final : public luci::CircleNodeMutableVisitor<void>
65 luci::CircleGatherNd gather_nd_node;
68 ASSERT_THROW(gather_nd_node.accept(&tv), std::exception);
71 TEST(CircleGatherNdTest, visit_NEG)
73 struct TestVisitor final : public luci::CircleNodeVisitor<void>
77 luci::CircleGatherNd gather_nd_node;
80 ASSERT_THROW(gather_nd_node.accept(&tv), std::exception);