Simulate UCLA games

sim.rb

#!/usr/bin/env ruby

require 'csv'
require 'pp'

$season = {}

# Data from http://www.spreadsheet-sports.com/2014-ncaa-tournament-data/
CSV.foreach('2014 Game Results Data.csv', headers: true, return_headers: true) do |game|
  $season[game["Team"]] ||= []<< game
  $season[game["Team"]] << game
end

def play(team_a, team_b)
  team_a_game_result = $season[team_a].sample["Team Result"]
  team_b_game_result = $season[team_b].sample["Team Result"]

  #  puts "Team: #{team_a} - #{team_a_game_result}"
  #  pp $season[team_a]

  #  puts "Team: #{team_b} - #{team_b_game_result}"
  #  pp $season[team_b]

  if team_a_game_result == "Win" && team_b_game_result == "Loss"
    team_a
  elsif team_a_game_result == "Loss" && team_b_game_result == "Win"
    team_b
  else
    rand > 0.5 ? team_a : team_b
  end                    
end                      
                         
team_a = ARGV.shift      
team_b = ARGV.shift      
                         
results = (0..100).map { play(team_a, team_b) }
h = Hash.new(0)
results.each { | v | h.store(v, h[v]+1) }
pp h