Skip to content

GitLab

Commit be87cc98 authored by Duc's avatar Duc
Browse files

added simple test

parent 2e57b128
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
test/test_fps.py
View file @ be87cc98
...@@ -21,6 +21,9 @@ def test_fps(dtype, device): ...@@ -21,6 +21,9 @@ def test_fps(dtype, device):
], dtype, device) ], dtype, device)
batch = tensor([0, 0, 0, 0, 1, 1, 1, 1], torch.long, device) batch = tensor([0, 0, 0, 0, 1, 1, 1, 1], torch.long, device)
out = fps(x, batch, ratio=0.5, random_start=False)
assert out.tolist() == [0, 2, 4, 6]
out = fps(x, batch, ratio=torch.tensor(0.5), random_start=False) out = fps(x, batch, ratio=torch.tensor(0.5), random_start=False)
assert out.tolist() == [0, 2, 4, 6] assert out.tolist() == [0, 2, 4, 6]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment