Imported Upstream version 1.7.0
[platform/core/ml/nnfw.git] / compiler / loco / src / IR / Verifier.test.cpp
1 /*
2  * Copyright (c) 2019 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 "loco/IR/Verifier.h"
18
19 #include <gtest/gtest.h>
20
21 #include <stdex/Memory.h>
22 #include <vector>
23
24 using stdex::make_unique;
25
26 TEST(VerifierTest, valid_minimal)
27 {
28   auto g = loco::make_graph();
29   auto push = g->nodes()->create<loco::Push>();
30
31   ASSERT_FALSE(loco::valid(g.get()));
32 }
33
34 TEST(VerifierTest, valid_error_reporter)
35 {
36   using namespace loco;
37
38   auto g = loco::make_graph();
39   auto push = g->nodes()->create<loco::Push>();
40
41   class Collector final : public loco::ErrorListener
42   {
43   public:
44     Collector(std::vector<ErrorDetail<ErrorCategory::MissingArgument>> *out) : _out{out}
45     {
46       // DO NOTHING
47     }
48
49   public:
50     void notify(const ErrorDetail<ErrorCategory::MissingArgument> &d) override
51     {
52       _out->emplace_back(d);
53     }
54
55   private:
56     std::vector<ErrorDetail<ErrorCategory::MissingArgument>> *_out;
57   };
58
59   std::vector<ErrorDetail<ErrorCategory::MissingArgument>> errors;
60   ASSERT_FALSE(loco::valid(g.get(), make_unique<Collector>(&errors)));
61   ASSERT_EQ(1, errors.size());
62   ASSERT_EQ(push, errors.at(0).node());
63   ASSERT_EQ(0, errors.at(0).index());
64 }