@@ -19,7 +19,7 @@ AIPrimTest >> setUp [
19
19
{ #category : ' tests' }
20
20
AIPrimTest >> testMinSpanningTreeComplex2 [
21
21
22
- | tree graphType graph |
22
+ | resultEdges graphType graph |
23
23
graphType := AICyclicWeightedComplexFixture new .
24
24
graph := graphType complexWeightedGraph4.
25
25
prim nodes: graph nodes.
@@ -29,17 +29,17 @@ AIPrimTest >> testMinSpanningTreeComplex2 [
29
29
to: #second
30
30
weight: #third .
31
31
prim run.
32
- tree := prim reconstructMST collect: [ :e | e asTuple ].
33
- self assert: tree size equals: 11 .
32
+ resultEdges := prim minimumSpanningTreeEdges collect: [ :e | e asTuple ].
33
+ self assert: resultEdges size equals: 11 .
34
34
self
35
- assert: (tree inject: 0 into: [ :sum :edge | sum + edge third ])
35
+ assert: (resultEdges inject: 0 into: [ :sum :edge | sum + edge third ])
36
36
equals: 112
37
37
]
38
38
39
39
{ #category : ' tests' }
40
40
AIPrimTest >> testMinimumSpanningTreeComplex [
41
41
42
- | tree expectedEdges graphType graph |
42
+ | resultEdges expectedEdges graphType graph |
43
43
graphType := AICyclicWeightedComplexFixture new .
44
44
graph := graphType complexWeightedGraph3.
45
45
prim nodes: graph nodes.
@@ -49,20 +49,20 @@ AIPrimTest >> testMinimumSpanningTreeComplex [
49
49
to: #second
50
50
weight: #third .
51
51
prim run.
52
- tree := prim reconstructMST collect: [ :e | e asTuple ].
52
+ resultEdges := prim minimumSpanningTreeEdges collect: [ :e | e asTuple ].
53
53
54
54
expectedEdges := #( #( $a $d 1 ) #( $d $e 1 ) #( $d $b 2 )
55
55
#( $e $c 5 ) ).
56
- self assertCollection: tree hasSameElements: expectedEdges.
56
+ self assertCollection: resultEdges hasSameElements: expectedEdges.
57
57
self
58
- assert: (tree inject: 0 into: [ :sum :edge | sum + edge third ])
58
+ assert: (resultEdges inject: 0 into: [ :sum :edge | sum + edge third ])
59
59
equals: 9
60
60
]
61
61
62
62
{ #category : ' tests' }
63
63
AIPrimTest >> testMinimumSpanningTreeSimple [
64
64
65
- | tree expectedEdges graphType graph |
65
+ | resultEdges expectedEdges graphType graph |
66
66
graphType := AICyclicWeightedSimpleFixture new .
67
67
graph := graphType simpleWeightedGraph2.
68
68
prim nodes: graph nodes.
@@ -72,11 +72,11 @@ AIPrimTest >> testMinimumSpanningTreeSimple [
72
72
to: #second
73
73
weight: #third .
74
74
prim run.
75
- tree := prim reconstructMST collect: [ :e | e asTuple ].
75
+ resultEdges := prim minimumSpanningTreeEdges collect: [ :e | e asTuple ].
76
76
77
77
expectedEdges := #( #( 1 2 3 ) #( 2 3 1 ) #( 3 4 2 ) #( 4 5 3 ) ).
78
- self assertCollection: tree hasSameElements: expectedEdges.
78
+ self assertCollection: resultEdges hasSameElements: expectedEdges.
79
79
self
80
- assert: (tree inject: 0 into: [ :sum :edge | sum + edge third ])
80
+ assert: (resultEdges inject: 0 into: [ :sum :edge | sum + edge third ])
81
81
equals: 9
82
82
]
0 commit comments