1 files changed, 45 insertions, 0 deletions
diff --git a/benchmarks/case.rb b/benchmarks/case.rb
new file mode 100644
index 0000000..2c51d0f
--- /dev/null
+++ b/benchmarks/case.rb
@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'benchmark'
+require 'ruby-enum'
+
+##
+# Test enum
+class Color
+  include Ruby::Enum
+  include Ruby::Enum::Case
+
+  define :RED, :red
+  define :GREEN, :green
+  define :BLUE, :blue
+end
+
+puts 'Running 1.000.000 normal case statements'
+case_statement_time = Benchmark.realtime do
+  1_000_000.times do
+    case Color::RED
+    when Color::RED, Color::GREEN
+      'red or green'
+    when Color::BLUE
+      'blue'
+    end
+  end
+end
+
+puts 'Running 1.000.000 ruby-enum case statements'
+ruby_enum_time = Benchmark.realtime do
+  1_000_000.times do
+    Color.case(Color::RED,
+               {
+                 [Color::RED, Color::GREEN] => -> { 'red or green' },
+                 Color::BLUE => -> { 'blue' }
+               })
+  end
+end
+
+puts "ruby-enum case: #{ruby_enum_time.round(4)}"
+puts "case statement: #{case_statement_time.round(4)}"
+
+puts "ruby-enum case is #{(ruby_enum_time / case_statement_time).round(2)} times slower"