use BloomFilter::ConfigurationMismatch instead of ArgumentError

igrigorik · Apr 27, 2012 · 76c1dcd · 76c1dcd
1 parent ba20cc9
commit 76c1dcd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/lib/bloomfilter/native.rb b/lib/bloomfilter/native.rb
@@ -1,7 +1,9 @@
 module BloomFilter
+  BloomFilter::ConfigurationMismatch = Class.new(ArgumentError)
+
   class Native < Filter
     attr_reader :bf
-
+    
     def initialize(opts = {})
       @opts = {
         :size    => 100,
@@ -45,7 +47,7 @@ def set_bits
     # It assumes that both filters have the same size -
     # if this is not true +ArgumentError+ is raised.
     def &(o)
-      raise ArgumentError.new() unless same_parameters?(o)
+      raise BloomFilter::ConfigurationMismatch.new unless same_parameters?(o)
       result = self.class.new
       result.instance_variable_set(:@bf,@bf.&(o.bf))
       result
@@ -55,7 +57,7 @@ def &(o)
     # It assumes that both filters have the same size -
     # if this is not true +ArgumentError+ is raised.
     def |(o)
-      raise ArgumentError.new() unless same_parameters?(o)
+      raise BloomFilter::ConfigurationMismatch.new unless same_parameters?(o)
       result = self.class.new
       result.instance_variable_set(:@bf,@bf.|(o.bf))
       result

diff --git a/spec/native_spec.rb b/spec/native_spec.rb
@@ -78,7 +78,7 @@
       bf2 = Native.new(:size => 20)
       bf2.insert("test")
 
-      proc {bf1 & bf2}.should raise_error(ArgumentError)
+      proc {bf1 & bf2}.should raise_error(BloomFilter::ConfigurationMismatch)
     end
 
     it "should return union with other filter" do