1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include <boost/test/unit_test.hpp>
7
8 #include "Graph.hpp"
9 #include "Observable.hpp"
10
11 BOOST_AUTO_TEST_SUITE(Observable)
12
BOOST_AUTO_TEST_CASE(AddedLayerObservableTest)13 BOOST_AUTO_TEST_CASE(AddedLayerObservableTest)
14 {
15 armnn::Graph graph;
16
17 // Create a graph observable
18 armnn::AddedLayerObservable layerObservable(graph);
19
20 // Add a few layers
21 auto output = graph.AddLayer<armnn::OutputLayer>(0, "output");
22 auto input = graph.InsertNewLayer<armnn::InputLayer>(output->GetInputSlot(0), 0, "input");
23
24 // Check the observable has observed the changes
25 std::list<armnn::Layer*> testLayers({ output, input });
26
27 BOOST_CHECK_EQUAL_COLLECTIONS(layerObservable.begin(), layerObservable.end(),
28 testLayers.begin(), testLayers.end());
29 }
30
BOOST_AUTO_TEST_CASE(ClearAddedLayerObservableTest)31 BOOST_AUTO_TEST_CASE(ClearAddedLayerObservableTest)
32 {
33 armnn::Graph graph;
34
35 // Create a graph observable
36 armnn::AddedLayerObservable addedLayerObservable(graph);
37
38 // Add a few layers
39 auto output = graph.AddLayer<armnn::OutputLayer>(0, "output");
40 graph.InsertNewLayer<armnn::InputLayer>(output->GetInputSlot(0), 0, "input");
41
42 addedLayerObservable.Clear();
43
44 // Check the observable has observed the changes
45 std::list<armnn::Layer*> emptyList({});
46
47 BOOST_CHECK_EQUAL_COLLECTIONS(addedLayerObservable.begin(), addedLayerObservable.end(),
48 emptyList.begin(), emptyList.end());
49 }
50
BOOST_AUTO_TEST_CASE(ErasedLayerNamesObservableTest)51 BOOST_AUTO_TEST_CASE(ErasedLayerNamesObservableTest)
52 {
53 armnn::Graph graph;
54
55 // Create a graph observable
56 armnn::ErasedLayerNamesObservable erasedLayerNamesObservable(graph);
57
58 // Add a few layers
59 auto output = graph.AddLayer<armnn::OutputLayer>(0, "output");
60 graph.InsertNewLayer<armnn::InputLayer>(output->GetInputSlot(0), 0, "input");
61
62 graph.EraseLayer(output);
63
64 // Check the observable has observed the changes
65 std::list<std::string> testList({"output"});
66
67 BOOST_CHECK_EQUAL_COLLECTIONS(erasedLayerNamesObservable.begin(), erasedLayerNamesObservable.end(),
68 testList.begin(), testList.end());
69 }
70
BOOST_AUTO_TEST_CASE(ClearErasedLayerNamesObservableTest)71 BOOST_AUTO_TEST_CASE(ClearErasedLayerNamesObservableTest)
72 {
73 armnn::Graph graph;
74
75 // Create a graph observable
76 armnn::ErasedLayerNamesObservable erasedLayerNamesObservable(graph);
77
78 // Add a few layers
79 auto output = graph.AddLayer<armnn::OutputLayer>(0, "output");
80 graph.InsertNewLayer<armnn::InputLayer>(output->GetInputSlot(0), 0, "input");
81
82 graph.EraseLayer(output);
83
84 erasedLayerNamesObservable.Clear();
85
86 // Check the observable has observed the changes
87 std::list<std::string> emptyList({});
88
89 BOOST_CHECK_EQUAL_COLLECTIONS(erasedLayerNamesObservable.begin(), erasedLayerNamesObservable.end(),
90 emptyList.begin(), emptyList.end());
91 }
92
93 BOOST_AUTO_TEST_SUITE_END()
94
95