From 3b132cc2203d4ad975c458ed13872cd3c98bad00 Mon Sep 17 00:00:00 2001
From: Alpha Chen <alpha.chen@gmail.com>
Date: Sun, 11 Dec 2016 11:23:00 -0500
Subject: [PATCH] [2016][ruby][11.0] implement equality on states

---
 2016/ruby/day_11.rb | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/2016/ruby/day_11.rb b/2016/ruby/day_11.rb
index 4572403..b12ee5a 100644
--- a/2016/ruby/day_11.rb
+++ b/2016/ruby/day_11.rb
@@ -10,6 +10,10 @@ class State
       Floor.new(items.reject {|item| item == ?. })
     }
   end
+
+  def ==(state)
+    elevator == state.elevator && floors.zip(state.floors).all? {|a,b| a == b }
+  end
 end
 
 class Floor < SimpleDelegator
@@ -51,4 +55,9 @@ F1 E  .  HM .  LM
     assert_equal %w[ H L ], floor.microchips
     assert_empty floor.generators
   end
+
+  def test_equality
+    state = State.new(INPUT)
+    assert_equal @state, state
+  end
 end