test no batch

Author: rusty1s

test/test_fps.py

@@ -23,3 +23,6 @@ def test_fps(dtype, device):
 
     out = fps(x, batch, ratio=0.5, random_start=False)
     assert out.tolist() == [0, 2, 4, 6]
+
+    out = fps(x, ratio=0.5, random_start=False)
+    assert out.sort()[0].tolist() == [0, 5, 6, 7]

test/test_nearest.py

@@ -31,3 +31,6 @@ def test_nearest(dtype, device):
 
     out = nearest(x, y, batch_x, batch_y)
     assert out.tolist() == [0, 0, 1, 1, 2, 2, 3, 3]
+
+    out = nearest(x, y)
+    assert out.tolist() == [0, 0, 1, 1, 2, 2, 3, 3]