def test_adding_neighbour():
v = Vertex(1, 1)
v.add_neighbour(2)
v.add_neighbour(3)
assert len(v.get_neighbours()) == 2
assert v.get_neighbours()[0] == 2
assert v.get_neighbours()[1] == 3
def test_cleaning_neighbours():
v = Vertex(1, 1)
v.add_neighbour(2)
v.add_neighbour(3)
v.clean_neighbours()
assert len(v.get_neighbours()) == 0
def test_removing_neighbour():
v = Vertex(1, 1)
v.add_neighbour(2)
v.add_neighbour(3)
v.remove_neighbour(2)
assert 2 not in v.get_neighbours()
