Type checking is added in Vector.coerce method

PlumLabs · Nov 4, 2008 · 89a290b · 89a290b
1 parent ecafddf
commit 89a290b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/vector.rb b/lib/vector.rb
@@ -39,8 +39,12 @@ def *(scalar)
       Vector.new(x * scalar, y * scalar)
     end
 
-    def coerce(scalar)
-      [self, scalar]
+    def coerce(scalar)      
+      if scalar.is_a?(Numeric)
+        [self, scalar]
+      else
+        raise ArgumentError, "Cannot coerce #{scalar.inspect}"
+      end             
     end
   end
 end