added unique test

Author: rusty1s

setup.py

@@ -2,7 +2,7 @@
 import torch
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension
 
-__version__ = '0.2.0'
+__version__ = '0.2.1'
 url = 'https://github.com/rusty1s/pytorch_sparse'
 
 install_requires = ['scipy']

test/test_coalesce.py

@@ -2,6 +2,15 @@
 from torch_sparse import coalesce
 
 
+def test_coalesce():
+    row = torch.tensor([1, 0, 1, 0, 2, 1])
+    col = torch.tensor([0, 1, 1, 1, 0, 0])
+    index = torch.stack([row, col], dim=0)
+
+    index, _ = coalesce(index, None, m=3, n=2)
+    assert index.tolist() == [[0, 1, 1, 2], [1, 0, 1, 0]]
+
+
 def test_coalesce_add():
     row = torch.tensor([1, 0, 1, 0, 2, 1])
     col = torch.tensor([0, 1, 1, 1, 0, 0])

torch_sparse/__init__.py

@@ -4,7 +4,7 @@
 from .spmm import spmm
 from .spspmm import spspmm
 
-__version__ = '0.2.0'
+__version__ = '0.2.1'
 
 __all__ = [
     '__version__',