From 796045f5a31b194f491c99f0258ac9a81b701fe1 Mon Sep 17 00:00:00 2001 From: Dmytro Mishkin Date: Fri, 11 Aug 2023 11:20:12 +0200 Subject: [PATCH] test actually works (#2510) --- test/feature/test_matching.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/test/feature/test_matching.py b/test/feature/test_matching.py index 25c15c7e08..209d57f85a 100644 --- a/test/feature/test_matching.py +++ b/test/feature/test_matching.py @@ -468,11 +468,8 @@ def test_real(self, device, dtype, data): assert idxs.shape[0] == dists.shape[0] assert dists.shape[0] <= data_dev['descs1'].shape[0] assert dists.shape[0] <= data_dev['descs2'].shape[0] - # key doesn't exist in data_dev - # expected_idxs = data_dev['lightglue_disk_idxs'].long() - data_dev['expected_idxs'].long() - # TODO: fix this - # assert_close(idxs, expected_idxs, rtol=1e-4, atol=1e-4) + expected_idxs = data_dev['lightglue_disk_idxs'].long() + assert_close(idxs, expected_idxs, rtol=1e-4, atol=1e-4) @pytest.mark.parametrize("data", ["lightglue_idxs"], indirect=True) def test_single_nocrash(self, device, dtype, data):