Add testcases for calling results to empty votes.

[rubyvote] / test / irv_test.rb

diff --git a/test/irv_test.rb b/test/irv_test.rb
index 49aa540e02a958bb1220ef22f47987e7d227b64d..fa05dc99bd62ebef470826eec4bad690a1b6d94b 100644 (file)
--- a/test/irv_test.rb
+++ b/test/irv_test.rb
@@ -6,6 +6,11 @@ require 'rubyvote/irv'
 
 class TestRunoffVote < Test::Unit::TestCase
 
+  def test_irv_empty
+    vote_array = Array.new
+    assert_nil InstantRunoffVote.new(vote_array).result.winners[0]
+  end
+  
   def test_irv
     vote_array = Array.new
     142.times {vote_array << "ABCD".split("")}
@@ -50,7 +55,12 @@ class TestRunoffVote < Test::Unit::TestCase
 
     assert_equal( "C", InstantRunoffVote.new(vote_array).result.winners[0] )
   end
-
+  
+  def test_irv_logic_empty
+    vote_array = Array.new
+    assert_nil InstantRunoffLogicVote.new(vote_array).result.winners[0]
+  end
+  
   def test_irv_logic1
     vote_array = Array.new
     42.times {vote_array << "ABCD".split("")}