Tidied up the let clauses in edge_spec

[graph.njae.git] / spec / graph / edge_spec.rb

diff --git a/spec/graph/edge_spec.rb b/spec/graph/edge_spec.rb
index 8fc71ba76c5bfa42b7ee3843e32d144c3a111e5a..bd823bfd919ab882fed7d7267710413018b648e6 100644 (file)
--- a/spec/graph/edge_spec.rb
+++ b/spec/graph/edge_spec.rb
@@ -3,6 +3,10 @@ require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
 module GraphNjae
   describe Edge do
     let (:e) { Edge.new }
+    let(:v1) {Vertex.new :name => :v1}
+    let(:v2) {Vertex.new :name => :v2}
+    let(:v3) {Vertex.new :name => :v3}
+
     describe "#initialize" do
       it "creates an empty edge" do
         e = Edge.new
@@ -28,8 +32,6 @@ module GraphNjae
     describe "#<<" do
       it "adds a new vertex to an edge (with a connection)" do
         e.connections.should be_empty
-        v1 = Vertex.new :name => :v1
-        v2 = Vertex.new :name => :v2
         e << v1
         e.should have(1).connections
         e.should have(1).vertices
@@ -45,8 +47,6 @@ module GraphNjae
       
       it "adds several vertices to an edge" do
         e.connections.should be_empty
-        v1 = Vertex.new :name => :v1
-        v2 = Vertex.new :name => :v2
         e << v1 << v2
         e.vertices.should include(v1)
         e.vertices.should include(v2)
@@ -55,7 +55,6 @@ module GraphNjae
 
       it "adds a self-loop" do
         e.connections.should be_empty
-        v1 = Vertex.new
         e << v1 << v1
         e.vertices.should include(v1)
         e.should have(2).vertices
@@ -66,8 +65,6 @@ module GraphNjae
     describe "connection_at" do
       it "returns the connection that links to a vertex" do
         e.connections.should be_empty
-        v1 = Vertex.new :name => :v1
-        v2 = Vertex.new :name => :v2
         e << v1 << v2
         
         e.connection_at(v1).end.should be v1
@@ -76,9 +73,6 @@ module GraphNjae
       
       it "returns nil if there is no connection to that vertex" do
         e.connections.should be_empty
-        v1 = Vertex.new :name => :v1
-        v2 = Vertex.new :name => :v2
-        v3 = Vertex.new :name => :v3
         e << v1 << v2
         
         e.connection_at(v3).should be nil
@@ -86,7 +80,6 @@ module GraphNjae
       
       it "returns the vertex for a self-loop" do
         e.connections.should be_empty
-        v1 = Vertex.new :name => :v1
         e << v1 << v1
         
         e.connection_at(v1).end.should be v1
@@ -96,8 +89,6 @@ module GraphNjae
     describe "other_end" do
       it "returns the vertex at the other end of the given one" do
         e.connections.should be_empty
-        v1 = Vertex.new :name => :v1
-        v2 = Vertex.new :name => :v2
         e << v1 << v2
         
         e.other_end(v1).should be v2
@@ -106,7 +97,6 @@ module GraphNjae
       
       it "returns the same vertex in a self-loop" do
         e.connections.should be_empty
-        v1 = Vertex.new :name => :v1
         e << v1 << v1
         
         e.other_end(v1).should be v1
@@ -114,21 +104,38 @@ module GraphNjae
       
       it "returns one of the connected edges if given a vertex not connected to it" do
         e.connections.should be_empty
-        v1 = Vertex.new :name => :v1
-        v2 = Vertex.new :name => :v2
-        v3 = Vertex.new :name => :v3
         e << v1 << v2
         
         [v1, v2].should include e.other_end(v3)
       end 
       
       it "returns nil if it can't return something sensible" do
-        v1 = Vertex.new :name => :v1
         e.other_end(v1).should be_nil
         e << v1
         e.other_end(v1).should be_nil
       end
     end # other_end
+    
+    describe "#to_dot" do
+      it "describes an edge in dot notation" do
+        e << v1 << v2
+
+        e.to_dot.should == "#{v1.object_id.to_s} -- #{v2.object_id.to_s};"
+      end
+      
+      it "describes an edge in dot notation, using given attributes" do
+        e << v1 << v2
+        #vdot = v1.to_dot :label => :name, :shape => :shape
+        #vdot.should == "#{v.object_id.to_s} {label = \"vertex\", shape = \"house\"};"
+      end
+
+      
+      it "describes an edge in dot notation, given a block" do
+        e << v1 << v2
+
+        e.to_dot {|e| e.object_id.to_s}.should == e.object_id.to_s
+      end
+    end # dot
   end # Edge
   
   describe Connection do