WIP #998 Added test case

deepforge-dev · Mar 31, 2017 · 03fe2f4 · 03fe2f4
1 parent 180bad4
commit 03fe2f4
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/src/seeds/devTests/devTests.webgmex b/src/seeds/devTests/devTests.webgmex
diff --git a/test/plugins/GenerateArchitecture/GenerateArchitecture.spec.js b/test/plugins/GenerateArchitecture/GenerateArchitecture.spec.js
@@ -90,6 +90,7 @@ describe('GenerateArchitecture', function () {
             //['/8', 'basic-transfers.lua'],
             //['/M', 'concat-parallel.lua'],
             ['/Q', 'basiccontainer.lua'],
+            ['/P', 'ContainerWithLayerArgs.lua'],
             ['/4', 'requiredOmitted.lua'],
             ['/e', 'googlenet.lua'],
             ['/X', 'overfeat.lua']

diff --git a/test/test-cases/generated-code/ContainerWithLayerArgs.lua b/test/test-cases/generated-code/ContainerWithLayerArgs.lua
@@ -0,0 +1,13 @@
+require 'nn'
+require 'rnn'
+
+local m = nn.Sequential()
+m:add(nn.Linear(100, 50))
+m:add(nn.LeakyReLU())
+m:add(nn.Linear(50, 10))
+
+
+local net = nn.Sequential()
+net:add(nn.Bottle(m, 2))
+
+return net