Imported Upstream version 1.7.0
[platform/core/ml/nnfw.git] / compiler / nest / core / src / expr / VarNode.test.cpp
1 /*
2  * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
3  *
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
7  *
8  *    http://www.apache.org/licenses/LICENSE-2.0
9  *
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.
15  */
16
17 #include "nest/expr/VarNode.h"
18
19 #include <memory>
20
21 #include <gtest/gtest.h>
22
23 TEST(VAR_NODE, ctor)
24 {
25   auto make = [](uint32_t n) {
26     const nest::VarID id{n};
27
28     return std::make_shared<nest::expr::VarNode>(id);
29   };
30
31   auto node = make(4);
32
33   // NOTE 'id' should be copied
34   ASSERT_EQ(4, node->id().value());
35 }
36
37 TEST(VAR_NODE, cast)
38 {
39   const nest::VarID id{0};
40
41   auto derived = std::make_shared<nest::expr::VarNode>(id);
42   std::shared_ptr<nest::expr::Node> base = derived;
43
44   // NOTE Cast method should be overrided
45   ASSERT_NE(derived.get(), nullptr);
46   ASSERT_EQ(derived.get(), base->asVar());
47 }