1 files changed, 28 insertions, 0 deletions
diff --git a/test/test_JustElement.lua b/test/test_JustElement.lua
new file mode 100644
index 0000000..d6c49a8
--- /dev/null
+++ b/test/test_JustElement.lua
@@ -0,0 +1,28 @@
+
+require 'totem'
+require 'nngraph'
+local test = {}
+local tester = totem.Tester()
+
+function test.test_output()
+   local input = {torch.randn(7, 11)}
+   local module = nngraph.JustElement()
+   tester:eq(module:forward(input), input[1], "output")
+end
+
+function test.test_grad()
+   local input = {torch.randn(7, 11)}
+   local module = nngraph.JustElement()
+   totem.nn.checkGradients(tester, module, input)
+end
+
+function test.test_split()
+   local in1 = nn.Identity()()
+   local output = in1:split(1)
+   local net = nn.gModule({in1}, {output})
+
+   local input = {torch.randn(7, 11)}
+   tester:eq(net:forward(input), input[1], "output of split(1)")
+end
+
+tester:add(test):run()