def test_one_big_neighborhood(self):
dummy_line_segs = [self.DummyTrajLineSeg(4)] * 4
expected = 2.0
result = find_entropy(dummy_line_segs)
self.assertEquals(expected, result)
def test_different_amount_of_neighbors(self):
dummy_line_segs = [self.DummyTrajLineSeg(0)] * 2
dummy_line_segs.append(self.DummyTrajLineSeg(1))
expected = 1.5
result = find_entropy(dummy_line_segs)
self.assertEquals(expected, result)
def test_zero_neighbors_each(self):
dummy_line_segs = [self.DummyTrajLineSeg(0)] * 8
expected = 3.0
result = find_entropy(dummy_line_segs)
self.assertEquals(expected, result)