Ejemplo n.º 1
0
        public void AddNodeTest()
        {
            var model = new StaticModel();

            Assert.NotNull(model.GetLayerNodes());
            Assert.That(model.GetLayerNodes()?.Length == 0);

            var layerType = new LayerType("Dropout", "Core",
                                          new[]
            {
                new Parameter("rate", "Float", "1.0", Array.Empty <string>())
            }
                                          );

            model.AddNode(layerType);

            Assert.That(model.GetLayerNodes()?.Length == 1);
        }
Ejemplo n.º 2
0
        public void MoveNodeTest()
        {
            var model = new StaticModel();

            Assert.NotNull(model.GetLayerNodes());
            Assert.That(model.GetLayerNodes()?.Length == 0);

            var layerType = new LayerType("Dropout", "Core",
                                          new[]
            {
                new Parameter("rate", "Float", "1.0", Array.Empty <string>())
            }
                                          );

            model.AddNode(layerType);
            var firstNode = model.GetLayerNodes()?.First();

            Assert.AreEqual(firstNode, model.GetLayerNodes()?[0]);

            model.AddNode(layerType);
            model.MoveNode(1, 0);

            Assert.AreEqual(firstNode, model.GetLayerNodes()?[1]);

            model.MoveNode(0, 1);

            Assert.AreEqual(firstNode, model.GetLayerNodes()?[0]);
        }