def test_init(self): w = d.KNN(self.euclidean_kdt, k=2) self.assertEqual(w.neighbors[0], [1, 3])
def test_reweight(self): w = d.KNN(self.points, k=2) new_point = [(21, 21)] wnew = w.reweight(k=4, p=1, new_data=new_point, inplace=False) self.assertEqual(wnew[0], {1: 1.0, 3: 1.0, 4: 1.0, 6: 1.0})