two_arg_expression.rb in gobstones-0.0.2

- old
+ new

@@ -1,11 +1,15 @@
+require 'gobstones/modules/equal_by_class'
+
 module Gobstones
 
   module Lang
 
     class TwoArgExpression
 
+      include Gobstones::EqualByClass
+
       attr_reader :left_expr, :right_expr
 
       def self.evaluates_with(selector)
         instance_eval do
           define_method :evaluate do |*args|
@@ -20,12 +24,12 @@
         @left_expr = left
         @right_expr = right
       end
 
       def ==(other)
-        self.class == other.class &&
-          self.left_expr == other.left_expr &&
-          self.right_expr == other.right_expr
+        super(other) &&
+        self.left_expr == other.left_expr &&
+        self.right_expr == other.right_expr
       end
 
     end
 
   end