1
1
from bzrlib.selftest import TestCase
2
from bzrlib.graph import farthest_nodes
2
from bzrlib.graph import node_distances, nodes_by_distance
4
4
class TestBase(TestCase):
5
5
def edge_add(self, *args):
18
18
self.edge_add('A', 'G', 'H', 'I', 'B')
19
19
self.edge_add('A', 'J', 'K', 'L', 'M', 'N')
20
20
self.edge_add('O', 'N')
22
def test_farthest(self):
22
def node_descendants(self):
23
23
descendants = {'A':set()}
24
24
for node in self.graph:
25
25
for ancestor in self.graph[node]:
26
26
if ancestor not in descendants:
27
27
descendants[ancestor] = set()
28
28
descendants[ancestor].add(node)
29
nodes = farthest_nodes(self.graph, descendants, 'A')
31
def test_distances(self):
32
descendants = self.node_descendants()
33
distances = node_distances(self.graph, descendants, 'A')
34
nodes = nodes_by_distance(distances)
30
35
self.assertEqual(nodes[0], 'D')
31
36
assert nodes[1] in ('N', 'C')
32
37
assert nodes[2] in ('N', 'C')
33
38
assert nodes[3] in ('B', 'M')
34
39
assert nodes[4] in ('B', 'M')
41
#Ensure we don't shortcut through B when there's only a difference of
44
self.edge_add('A', 'B', 'C')
45
self.edge_add('A', 'D', 'E', 'C')
46
descendants = self.node_descendants()
47
distances = node_distances(self.graph, descendants, 'A')
48
self.assertEqual(distances['C'], 3)
added
removed
bzrlib/selftest/testgraph.py
