type_balancer 0.1.0

data/benchmark/quick_benchmark.rb

@@ -0,0 +1,70 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'bundler/setup'
+require 'benchmark'
+require_relative '../lib/type_balancer'
+
+# Test cases of increasing size with different ratios
+TEST_CASES = [
+  { name: 'Tiny', total: 100, ratio: 0.1 },
+  { name: 'Small', total: 1_000, ratio: 0.2 },
+  { name: 'Medium', total: 10_000, ratio: 0.3 },
+  { name: 'Large', total: 100_000, ratio: 0.5 }
+].freeze
+
+# Print Ruby and YJIT information
+puts "Ruby version: #{RUBY_VERSION}"
+puts "YJIT enabled: #{RubyVM::YJIT.enabled? rescue false}"
+puts
+
+# Generate available items for each test case
+TEST_CASES.each do |test|
+  # Create available items as 20% more than needed
+  target = (test[:total] * test[:ratio]).round
+  available = (target * 1.2).round
+  test[:available] = Array.new(available) { |i| { id: i, type: 'video', title: "Item #{i}" } }
+end
+
+def run_benchmark(test_case)
+  puts "\nRunning benchmark for #{test_case[:name]} dataset:"
+  puts "- Total count: #{test_case[:total]}"
+  puts "- Available items: #{test_case[:available].size}"
+  puts "- Ratio: #{test_case[:ratio]}"
+  puts
+
+  # Adjust iterations based on dataset size
+  iterations = case test_case[:name]
+              when 'Tiny' then 100_000
+              when 'Small' then 10_000
+              when 'Medium' then 1_000
+              when 'Large' then 100
+              end
+
+  # Warmup phase
+  puts "Warming up..."
+  warmup_iterations = iterations / 10
+  warmup_iterations.times do
+    TypeBalancer.balance(test_case[:available], type_field: :type)
+  end
+  puts "Warmup complete."
+  puts
+
+  # Run benchmark
+  result = Benchmark.bm(20) do |x|
+    x.report("Ruby:") do
+      iterations.times do
+        TypeBalancer.balance(test_case[:available], type_field: :type)
+      end
+    end
+  end
+
+  # Calculate and display operations per second
+  ops_per_second = iterations / result[0].real
+  puts "\nOperations per second: #{ops_per_second.round(2)}"
+end
+
+# Run benchmarks for each test case
+TEST_CASES.each do |test_case|
+  run_benchmark(test_case)
+end