Commit bf162c98 authored by jay's avatar jay
Browse files

fixes outlier detector tests for API change

parent d166c9e7

autocnet/matcher/tests/test_outlier_detector.py

+2 −2
Original line number Diff line number Diff line
@@ -19,7 +19,7 @@ class TestOutlierDetector(unittest.TestCase): 
                                    [1.25, 10.1, 2.3, 2.4, 1.2, 5.5, 5.7]]).T,

                          columns=['source_idx', 'destination_idx', 'distance'])



        mask = cpu_outlier_detector.distance_ratio(df, ratio=0.8)

        mask = cpu_outlier_detector.distance_ratio(None, df, ratio=0.8)

        self.assertEqual(2, sum(mask))



    def test_distance_ratio_unique(self):
@@ -28,7 +28,7 @@ class TestOutlierDetector(unittest.TestCase): 
        df = pd.DataFrame(data, columns=['source_image', 'source_idx',

                                         'destination_image', 'destination_idx',

                                         'distance'])

        mask = cpu_outlier_detector.distance_ratio(df, ratio=0.9)

        mask = cpu_outlier_detector.distance_ratio(None, df, ratio=0.9)

        self.assertTrue(mask.all() == False)



    def test_mirroring_test(self):