added new tests

Author: rusty1s

setup.py

@@ -2,7 +2,7 @@
 
 from setuptools import setup, find_packages
 
-__version__ = '0.2.3'
+__version__ = '0.3.0'
 url = 'https://github.com/rusty1s/pytorch_scatter'
 
 install_requires = ['cffi']

test/backward.json

@@ -1,56 +1,56 @@
 [
   {
     "name": "add",
-    "index": [[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]],
-    "input": [[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]],
-    "dim": 1,
+    "index": [2, 0, 1, 1, 0],
+    "input": [1, 2, 3, 4, 5],
+    "dim": 0,
     "fill_value": 0,
-    "grad": [[10, 20, 30, 40, 50, 60], [15, 25, 35, 45, 55, 65]],
-    "expected": [[50, 60, 50, 30, 40], [15, 15, 35, 35, 25]]
+    "grad": [4, 8, 6],
+    "expected": [6, 4, 8, 8, 4]
   },
   {
-    "name": "add",
-    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
-    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
+    "name": "sub",
+    "index": [2, 0, 1, 1, 0],
+    "input": [1, 2, 3, 4, 5],
     "dim": 0,
     "fill_value": 0,
-    "grad": [[10, 20], [15, 25]],
-    "expected": [[10, 20], [15, 25], [15, 25], [10, 20]]
+    "grad": [4, 8, 6],
+    "expected": [-6, -4, -8, -8, -4]
   },
   {
     "name": "mean",
-    "index": [[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]],
-    "input": [[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]],
-    "dim": 1,
+    "index": [2, 0, 1, 1, 0],
+    "input": [1, 2, 3, 4, 5],
+    "dim": 0,
     "fill_value": 0,
-    "grad": [[10, 20, 30, 40, 50, 60], [15, 25, 35, 45, 55, 65]],
-    "expected": [[50, 60, 50, 30, 40], [15, 15, 35, 35, 25]]
+    "grad": [4, 8, 6],
+    "expected": [6, 2, 4, 4, 2]
   },
   {
-    "name": "mean",
-    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
-    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
+    "name": "max",
+    "index": [2, 0, 1, 1, 0],
+    "input": [1, 2, 3, 4, 5],
     "dim": 0,
     "fill_value": 0,
-    "grad": [[10, 20], [15, 25]],
-    "expected": [[10, 20], [15, 25], [15, 25], [10, 20]]
+    "grad": [4, 8, 6],
+    "expected": [6, 0, 0, 8, 4]
   },
   {
-    "name": "max",
-    "index": [[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]],
-    "input": [[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]],
-    "dim": 1,
-    "fill_value": 0,
-    "grad": [[10, 20, 30, 40, 50, 60], [15, 25, 35, 45, 55, 65]],
-    "expected": [[50, 60, 0, 30, 40], [0, 15, 0, 35, 25]]
+    "name": "min",
+    "index": [2, 0, 1, 1, 0],
+    "input": [1, 2, 3, 4, 5],
+    "dim": 0,
+    "fill_value": 3,
+    "grad": [4, 8, 6],
+    "expected": [6, 4, 8, 0, 0]
   },
   {
-    "name": "max",
-    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
-    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
+    "name": "mul",
+    "index": [2, 0, 1, 1, 0],
+    "input": [1, 2, 3, 4, 5],
     "dim": 0,
-    "fill_value": 0,
-    "grad": [[10, 20], [15, 25]],
-    "expected": [[10, 0], [0, 25], [15, 0], [0, 20]]
+    "fill_value": 2,
+    "grad": [4, 8, 6],
+    "expected": [12, 40, 64, 48, 16]
   }
 ]

test/utils.py

@@ -3,6 +3,8 @@
 
 tensors = [t[:-4] for t in tensor_classes]
 tensors.remove('ShortTensor')  # TODO: PyTorch `atomicAdd` bug with short type.
+tensors.remove('ByteTensor')  # We cannot properly test unsigned values.
+tensors.remove('CharTensor')  # Overflow on gradient computations :(
 
 
 def Tensor(str, x):