Added a primitive test file for gaussianpyramid

1 files changed, 25 insertions, 0 deletions

diff --git a/test/test_gaussianpyramid.lua b/test/test_gaussianpyramid.lua
new file mode 100644
index 0000000..f0fc603
--- /dev/null
+++ b/test/test_gaussianpyramid.lua
@@ -0,0 +1,25 @@
+require 'totem'
+require 'image'
+
+
+local tester = totem.Tester()
+
+
+local tests = {}
+
+function tests.doesNotCrash()
+  -- Char, Short and Int tensors not supported.
+  types = {
+      'torch.ByteTensor',
+      'torch.FloatTensor',
+      'torch.DoubleTensor'
+  }
+  for _, type in ipairs(types) do
+    local output = unpack(image.gaussianpyramid(torch.rand(8, 8):type(type), {0.5}))
+    tester:assert(output:type() == type, 'Type ' .. type .. ' produces a different output.')
+  end
+end
+
+
+return tester:add(tests):run()
+