test one-dimensional positions

test/test_grid.py

@@ -7,6 +7,13 @@ from .utils import tensors, Tensor
 
 @pytest.mark.parametrize('tensor', tensors)
 def test_grid_cluster_cpu(tensor):
+    position = Tensor(tensor, [0, 9, 2, 8, 3])
+    size = torch.LongTensor([5])
+    expected = torch.LongTensor([0, 1, 0, 1, 0])
+
+    output = grid_cluster(position, size)
+    assert output.tolist() == expected.tolist()
+
     position = Tensor(tensor, [[0, 0], [9, 9], [2, 8], [2, 2], [8, 3]])
     size = torch.LongTensor([5, 5])
     expected = torch.LongTensor([0, 3, 1, 0, 2])