/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/selftest/testgraph.py

  • Committer: Robert Collins
  • Date: 2005-10-16 22:31:25 UTC
  • mto: This revision was merged to the branch mainline in revision 1458.
  • Revision ID: robertc@lifelesslap.robertcollins.net-20051016223125-26d4401cb94b7b82
Branch.relpath has been moved to WorkingTree.relpath.

WorkingTree no no longer takes an inventory, rather it takes an optional branch
parameter, and if None is given will open the branch at basedir implicitly.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/selftest/testgraph.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
from bzrlib.tests import TestCase
18
 
from bzrlib.deprecated_graph import node_distances, nodes_by_distance, Graph
19
 
 
 
1
from bzrlib.selftest import TestCase
 
2
from bzrlib.graph import node_distances, nodes_by_distance
20
3
 
21
4
class TestBase(TestCase):
22
 
 
23
5
    def edge_add(self, *args):
24
6
        for start, end in zip(args[:-1], args[1:]):
25
7
            if start not in self.graph:
45
27
                    descendants[ancestor] = set()
46
28
                descendants[ancestor].add(node)
47
29
        return descendants
48
 
 
 
30
    
49
31
    def test_distances(self):
50
32
        descendants = self.node_descendants()
51
33
        distances = node_distances(self.graph, descendants, 'A')
52
34
        nodes = nodes_by_distance(distances)
53
35
        self.assertEqual(nodes[0], 'D')
54
 
        self.assert_(nodes[1] in ('N', 'C'))
55
 
        self.assert_(nodes[2] in ('N', 'C'))
56
 
        self.assert_(nodes[3] in ('B', 'M'))
57
 
        self.assert_(nodes[4] in ('B', 'M'))
 
36
        assert nodes[1] in ('N', 'C')
 
37
        assert nodes[2] in ('N', 'C')
 
38
        assert nodes[3] in ('B', 'M')
 
39
        assert nodes[4] in ('B', 'M')
58
40
 
59
41
        #Ensure we don't shortcut through B when there's only a difference of
60
42
        # 1 in distance
65
47
        distances = node_distances(self.graph, descendants, 'A')
66
48
        self.assertEqual(distances['C'], 3)
67
49
 
68
 
 
69
 
class TestGraph(TestCase):
70
 
 
71
 
    def test_get_descendants(self):
72
 
        # Graph objects let you get a descendants graph in
73
 
        # node: {direct-children:distance} which contains
74
 
        # known children, including ghost children
75
 
        graph = Graph()
76
 
        graph.add_ghost('ghost')
77
 
        graph.add_node('rev1', ['ghost'])
78
 
        # check the result contains ghosts:
79
 
        self.assertEqual({'ghost': {'rev1': 1}, 'rev1': {}},
80
 
                         graph.get_descendants())