Pavel Patsey
71 lines
1.5 KiB
Python
71 lines
1.5 KiB
Python
from vpn_manager.cidr4_merge.cidr4_merger import find_neighbors, merge_nodes, solution
|
|
|
|
|
|
def test_true():
|
|
assert True
|
|
|
|
|
|
def test_merge_nodes():
|
|
assert merge_nodes((0, 32), (1, 32)) == ((0, 31), 0)
|
|
assert merge_nodes((0, 32), (2, 32)) == ((0, 30), 2)
|
|
assert merge_nodes((0, 32), (5, 32)) == ((0, 29), 6)
|
|
assert merge_nodes((3, 32), (4, 32)) == ((0, 29), 6)
|
|
assert merge_nodes((0, 32), (4, 30)) == ((0, 29), 3)
|
|
assert merge_nodes((0, 32), (6, 31)) == ((0, 29), 5)
|
|
|
|
|
|
def test_find_neighbors():
|
|
assert find_neighbors([(0, 32), (2, 32), (4, 32), (6, 32)]) == []
|
|
assert find_neighbors([(0, 32), (1, 32)]) == [(0, (0, 32), (1, 32))]
|
|
assert find_neighbors(
|
|
[
|
|
(0, 32),
|
|
(1, 32),
|
|
(6, 32),
|
|
(7, 32),
|
|
]
|
|
) == [
|
|
(0, (0, 32), (1, 32)),
|
|
(2, (6, 32), (7, 32)),
|
|
]
|
|
assert find_neighbors(
|
|
[
|
|
(0, 32),
|
|
(1, 32),
|
|
(6, 32),
|
|
]
|
|
) == [
|
|
(0, (0, 32), (1, 32)),
|
|
]
|
|
|
|
|
|
def test_solution():
|
|
assert solution(
|
|
[
|
|
(0, 32),
|
|
(3, 32),
|
|
(4, 32),
|
|
],
|
|
2,
|
|
) == ([(0, 30), (4, 32)], 2)
|
|
|
|
assert solution(
|
|
[
|
|
(0, 32),
|
|
(3, 32),
|
|
(4, 32),
|
|
(7, 32),
|
|
],
|
|
2,
|
|
) == ([(0, 29)], 4)
|
|
|
|
assert solution(
|
|
[
|
|
(0, 32),
|
|
(3, 32),
|
|
(4, 32),
|
|
(7, 32),
|
|
],
|
|
1,
|
|
) == ([(0, 29)], 4)
|