lib/maintain/value.rb in maintain-0.1.4 vs lib/maintain/value.rb in maintain-0.1.5

@@ -19,11 +19,10 @@
     def ==(value)
       compare_value == compare_value_for(value)
     end
 
     def ===(value)
-      raise "#{compare_value} #{compare_value_for(value)}"
       (compare_value == compare_value_for(value)) || super
     end
 
     def class
       value.class
@@ -90,13 +89,13 @@
       end
       super
     end
 
     def state_name_for(value)
-      if (value.is_a?(String) || value.is_a?(Symbol))
+      if value.to_s =~ /^\d+$/
+        @state.state_name_for(value.to_i)
+      elsif (value.is_a?(String) || value.is_a?(Symbol))
         @state.states.has_key?(value.to_sym) ? value.to_sym : nil
-      else
-        @state.state_name_for(value)
       end
     end
 
     def state_value_for(state, value)
       if (state.is_a?(String) || state.is_a?(Symbol)) && state_hash = @state.states[state.to_sym]
\ No newline at end of file