pokerstats 2.0.1

data/spec/hand_statistics_spec_helper.rb

@@ -0,0 +1,89 @@
+require 'bigdecimal'
+require 'bigdecimal/util'
+require File.expand_path(File.dirname(__FILE__) + '/../lib/pokerstats/hand_constants')
+include HandConstants
+  
+module HandStatisticsFactories
+  def sample_hand
+    @@sample_hand_result ||= {
+      :session_filename => "a/b/c",
+      :starting_at => Time.now,
+      :name => "PS0001",
+      :description => "This is a description",
+      :sb => "1.00".to_d,
+      :bb => "2.00".to_d,
+      :board => "AS KS QS JS TS",
+      :total_pot => "100.00".to_d,
+      :rake => "4.00".to_d,
+      :played_at => Time.now,
+      :tournament => "T123"
+    }
+  end
+  
+  def sample_player
+    @@sample_player_result ||= {
+      :screen_name => 'sample_player_screen_name', 
+      :seat => 1
+    }
+  end
+  
+  def next_sample_player hash={}
+    @@sample_player_count ||= 0
+    player = sample_player.clone
+    player[:screen_name] = player[:screen_name] + "_" + @@sample_player_count.to_s
+    @@sample_player_count += 1
+    player.merge(hash)
+  end
+  
+  def sample_action
+    @@sample_action_result ||= {
+      :state => :prelude,
+      :screen_name => sample_player[:screen_name],
+      :action => :pay,
+      :result => :pay,
+      :amount => 10,
+      :other_thing => :baz
+    }
+  end
+  
+  def register_street(street, hash={})
+    @stats.update_hand(hash.merge(:street => street))
+  end
+  
+  def register_bet(player, amount, hash={})
+    @stats.register_action player[:screen_name], 'bets', hash.merge(:result => :pay, :amount => amount)
+  end
+  
+  def register_post(player, amount, hash={})
+    @stats.register_action player[:screen_name], 'posts', hash.merge(:result => :post, :amount => amount)
+  end
+  
+  def register_ante(player, amount, hash={})
+    @stats.register_action player[:screen_name], 'antes', hash.merge(:result => :post, :amount => amount)
+  end
+  
+  def register_raise_to(player, amount, hash={})
+    @stats.register_action player[:screen_name], 'raises', hash.merge(:result => :pay_to, :amount => amount)
+  end
+  
+  def register_call(player, amount, hash={})
+    @stats.register_action player[:screen_name], 'calls', hash.merge(:result => :pay, :amount => amount)
+  end
+  
+  def register_check(player, hash={})
+    @stats.register_action player[:screen_name], 'checks', hash.merge(:result => :neutral)
+  end
+  
+  def register_fold(player, hash={})
+    @stats.register_action player[:screen_name], 'folds', hash.merge(:result => :neutral)
+  end
+  
+  def register_win(player, amount, hash={})
+    @stats.register_action player[:screen_name], 'wins', hash.merge(:result => :win, :amount => amount)
+  end
+  
+  def register_cards(player, data, hash={})
+    @stats.register_action player[:screen_name], 'shows', hash.merge(:result => :cards, :data => data)
+  end  
+end
+include HandStatisticsFactories

data/spec/player_statistics_spec.rb

@@ -0,0 +1,230 @@
+require 'rubygems'
+require 'activesupport'
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+require File.expand_path(File.dirname(__FILE__) + '/../lib/pokerstats/hand_statistics')
+require File.expand_path(File.dirname(__FILE__) + '/../lib/pokerstats/player_statistics')
+include Pokerstats
+
+TEST = [
+  {"andy" => {
+      :is_blind_attack_opportunity => false, 
+      :is_blind_attack_opportunity_taken => nil, 
+      :is_blind_defense_opportunity => false, 
+      :is_blind_defense_opportunity_taken => nil,
+      :is_cbet_opportunity => true, 
+      :is_cbet_opportunity_taken => true,
+      :posted => 1, 
+      :paid => 2, 
+      :won => 3, 
+      :cards => nil,
+      :preflop_passive => 1, 
+      :preflop_aggressive => 2, 
+      :postflop_passive => 3, 
+      :postflop_aggressive => 4, 
+      :is_pfr_opportunity => true, 
+      :is_pfr_opportunity_taken => true
+    },
+  "judi" => {
+      :is_blind_attack_opportunity => true, 
+      :is_blind_attack_opportunity_taken => false, 
+      :is_blind_defense_opportunity => true, 
+      :is_blind_defense_opportunity_taken => true,
+      :is_cbet_opportunity => true, 
+      :is_cbet_opportunity_taken => false,
+      :posted => 10, 
+      :paid => 20, 
+      :won => 30, 
+      :cards => nil,
+      :preflop_passive => 10, 
+      :preflop_aggressive => 20, 
+      :postflop_passive => 30, 
+      :postflop_aggressive => 40,
+      :is_pfr_opportunity => true, 
+      :is_pfr_opportunity_taken => false
+    }
+  },
+  {"andy" => {
+      :is_blind_attack_opportunity => false, 
+      :is_blind_attack_opportunity_taken => nil, 
+      :is_blind_defense_opportunity => false, 
+      :is_blind_defense_opportunity_taken => nil,
+      :is_cbet_opportunity => true, 
+      :is_cbet_opportunity_taken => false,
+      :posted => 2, 
+      :paid => 4, 
+      :won => 6, 
+      :cards => nil,
+      :preflop_passive => 2, 
+      :preflop_aggressive => 4, 
+      :postflop_passive => 6, 
+      :postflop_aggressive => 8,
+      :is_pfr_opportunity => true, 
+      :is_pfr_opportunity_taken => true
+    },
+  "judi" => {
+      :is_blind_attack_opportunity => true, 
+      :is_blind_attack_opportunity_taken => false, 
+      :is_blind_defense_opportunity => true, 
+      :is_blind_defense_opportunity_taken => true,
+      :is_cbet_opportunity => false, 
+      :is_cbet_opportunity_taken => nil,
+      :posted => 20, 
+      :paid => 30, 
+      :won => 40, 
+      :cards => nil,
+      :preflop_passive => 20, 
+      :preflop_aggressive => 40, 
+      :postflop_passive => 60, 
+      :postflop_aggressive => 80,
+      :is_pfr_opportunity => true, 
+      :is_pfr_opportunity_taken => false
+    }
+  },
+  {"andy" => {
+      :is_blind_attack_opportunity => false, 
+      :is_blind_attack_opportunity_taken => nil, 
+      :is_blind_defense_opportunity => true, 
+      :is_blind_defense_opportunity_taken => false,
+      :is_cbet_opportunity => false, 
+      :is_cbet_opportunity_taken => nil,
+      :posted => 3, 
+      :paid => 6, 
+      :won => 9, 
+      :cards => nil,
+      :preflop_passive => 3, 
+      :preflop_aggressive => 6, 
+      :postflop_passive => 9, 
+      :postflop_aggressive => 12,
+      :is_pfr_opportunity => true, 
+      :is_pfr_opportunity_taken => true
+    },    
+  "judi" => {
+      :is_blind_attack_opportunity => true, 
+      :is_blind_attack_opportunity_taken => true, 
+      :is_blind_defense_opportunity => true, 
+      :is_blind_defense_opportunity_taken => true,
+      :is_cbet_opportunity => false, 
+      :is_cbet_opportunity_taken => nil,
+      :posted => 30, 
+      :paid => 0, 
+      :won => 0, 
+      :cards => nil,
+      :preflop_passive => 30, 
+      :preflop_aggressive => 60, 
+      :postflop_passive => 90, 
+      :postflop_aggressive => 120,
+      :is_pfr_opportunity => true, 
+      :is_pfr_opportunity_taken => false
+    }
+  }
+]
+
+PS_RESULT = {
+  "andy" => {
+    :t_hands => 3,
+    :t_vpip => 3,
+    :t_posted => 6, 
+    :t_paid => 12,
+    :t_won => 18,
+    :t_blind_attack_opportunity => 0, 
+    :t_blind_attack_opportunity_taken => 0, 
+    :t_blind_defense_opportunity => 1, 
+    :t_blind_defense_opportunity_taken => 0,
+    :t_cbet_opportunity => 2, 
+    :t_cbet_opportunity_taken => 1,
+    :t_preflop_passive => 6, 
+    :t_preflop_aggressive => 12, 
+    :t_postflop_passive => 18, 
+    :t_postflop_aggressive => 24,
+    :t_pfr_opportunity => 3, 
+    :t_pfr_opportunity_taken => 3
+  },
+  "judi" => {
+    :t_hands => 3,
+    :t_vpip => 2,
+    :t_posted => 60, 
+    :t_paid => 50,
+    :t_won => 70,
+    :t_blind_attack_opportunity => 3, 
+    :t_blind_attack_opportunity_taken => 1, 
+    :t_blind_defense_opportunity => 3, 
+    :t_blind_defense_opportunity_taken => 3,
+    :t_cbet_opportunity => 1, 
+    :t_cbet_opportunity_taken => 0,
+    :t_preflop_passive => 60, 
+    :t_preflop_aggressive => 120, 
+    :t_postflop_passive => 180, 
+    :t_postflop_aggressive => 240,
+    :t_pfr_opportunity => 3, 
+    :t_pfr_opportunity_taken => 0
+  }
+}
+
+class AggregateTo
+  def initialize(player, stat, expected)
+    @player = player
+    @stat = stat
+    @expected = expected
+  end
+
+  def matches?(hash)
+    @actual = hash[@player][@stat]
+    # Satisfy expectation here. Return false or raise an error if it's not met.
+    return @actual == @expected
+  end
+
+  def failure_message
+    "expected report[#{@player.inspect}][#{@stat.inspect}] to aggregate to #{@expected.inspect}, got #{@actual.inspect}"
+  end
+
+  def negative_failure_message
+    "expected report[#{@player.inspect}][#{@stat.inspect}] not to aggregate to #{@expected.inspect}, but it did"
+  end
+end
+
+def should_correctly_aggregate(reports, stats = [])
+  reports.keys.each do |each_player|
+    stats.each do |each_stat|
+      reports.should AggregateTo.new(each_player, each_stat, PS_RESULT[each_player][each_stat])
+    end 
+  end
+end
+
+describe PlayerStatistics, "when aggregating statistics" do
+  before do
+    @player_statistics = PlayerStatistics.new
+    for each_test in TEST
+      @hand_statistics = HandStatistics.new
+      @hand_statistics.should_receive(:reports).and_return(each_test)
+      @player_statistics.record(@hand_statistics)
+    end
+    @reports = @player_statistics.reports
+  end
+  
+  it "should produce a report with data for each player in the reports" do
+    players = {}
+    for each_hand in TEST
+      for each_player in each_hand.keys
+        players[each_player] = true
+      end
+    end
+    @reports.should have(players.size).keys
+  end
+  
+  it "should correctly aggregate aggression for each player" do
+    should_correctly_aggregate @reports, [:t_preflop_passive, :t_preflop_aggressive, :t_postflop_passive, :t_postflop_aggressive]
+  end
+  
+  it 'should correctly aggregate blind attack statistics' do
+    should_correctly_aggregate @reports, [:t_blind_attack_opportunity, :t_blind_attack_opportunity_taken, :t_blind_defense_opportunity, :t_blind_defense_opportunity_taken]
+  end
+  it 'should correctly aggregate cash statistics' do
+    should_correctly_aggregate @reports, [:t_hands, :t_vpip, :t_posted, :t_paid, :t_won]
+  end
+  it 'should correctly aggregate continuation bet statistics' do
+    should_correctly_aggregate @reports, [:t_cbet_opportunity, :t_cbet_opportunity_taken]
+  end
+  it 'should correctly aggregate preflop raise statistics' do
+    should_correctly_aggregate @reports, [:t_pfr_opportunity, :t_pfr_opportunity_taken]
+  end
+end

data/spec/pokerstars_file_spec.rb

@@ -0,0 +1,160 @@
+require 'rubygems'
+require 'activesupport'
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+require File.expand_path(File.dirname(__FILE__) + '/../lib/pokerstats/hand_history')
+require File.expand_path(File.dirname(__FILE__) + '/../lib/pokerstats/pokerstars_file')
+include Pokerstats
+
+describe PokerstarsFile, "when opened on an empty file" do
+  it "should complain" do
+    lambda{@psfile = PokerstarsFile.open(File.dirname(__FILE__) + '/file_empty.txt')}.should raise_error
+  end
+end
+
+describe PokerstarsFile, "when opened on a single-hand file" do
+  ONE_HAND_NUMBER_OF_ENTRIES = 1
+    
+  ONE_HAND_ENTRY_NUMBER_OF_LINES = 50
+  ONE_HAND_ENTRY_FIRST_LINE = "PokerStars Game #24759011305: Tournament #139359808, $10+$1 Hold'em No Limit - Level I (10/20) - 2009/02/09 14:02:39 ET"
+  ONE_HAND_ENTRY_LAST_LINE = "Seat 9: MartinBOF84 folded before Flop (didn't bet)"
+
+  before do
+    @psfile = PokerstarsFile.open(File.dirname(__FILE__) + '/file_one_hand.txt')
+  end
+  
+  it "should be open" do
+    @psfile.should_not be_closed
+  end
+  
+  it "should not be eof" do
+    @psfile.should_not be_eof
+  end
+  
+  it "should be at position zero" do
+    @psfile.pos.should be_zero
+  end
+  
+  it "should read all the hands in the test file" do
+    @psfile.should have(ONE_HAND_NUMBER_OF_ENTRIES).entries
+  end
+  
+  it "should have an entry, answering to first, having the correct lines" do
+    @entry = @psfile.first
+    @entry.should have(ONE_HAND_ENTRY_NUMBER_OF_LINES).lines
+    @entry.lines[0].should == ONE_HAND_ENTRY_FIRST_LINE
+    @entry.lines[ONE_HAND_ENTRY_NUMBER_OF_LINES-1].should == ONE_HAND_ENTRY_LAST_LINE
+    @psfile.should be_closed
+  end
+  
+  it "should have an entry, answering to entries, having the correct lines" do
+    @entry = @psfile.entries.first
+    @entry.should have(ONE_HAND_ENTRY_NUMBER_OF_LINES).lines
+    @entry.lines.first.should == ONE_HAND_ENTRY_FIRST_LINE
+    @entry.lines.last.should == ONE_HAND_ENTRY_LAST_LINE
+  end
+
+  it "should be at the end of file after reading the entry" do
+    @psfile.first
+    @psfile.should be_eof
+  end
+end
+
+describe PokerstarsFile, "when opened on a file encoded in Latin-1, should transliterate properly to ASCII" do
+  LATIN1_LINE_INDEX = 8
+  LATIN1_LINE_TRANSLITERATED = "Seat 8: Gw\"unni (3000 in chips) "
+  
+  before do
+    @psfile = PokerstarsFile.open(File.dirname(__FILE__) + '/file_one_hand.txt')
+    @entry = @psfile.entries.first
+  end
+  
+  it "should properly transliterate the selected line" do
+    @entry.lines[LATIN1_LINE_INDEX].should == LATIN1_LINE_TRANSLITERATED
+  end
+end
+
+describe PokerstarsFile, "when opened on a multi-hand file" do
+  NUMBER_OF_ENTRIES = 12
+  
+  FIRST_ENTRY_NUMBER_OF_LINES = 47
+  FIRST_ENTRY_FIRST_LINE = "PokerStars Game #21650146783:  Hold'em No Limit ($0.25/$0.50) - 2008/10/31 17:14:44 ET"
+  FIRST_ENTRY_LAST_LINE = "Seat 9: ATTACCA folded before Flop (didn't bet)"
+
+  LAST_ENTRY_NUMBER_OF_LINES = 48
+  LAST_ENTRY_FIRST_LINE = "PokerStars Game #21650401569:  Hold'em No Limit ($0.25/$0.50) - 2008/10/31 17:24:58 ET"
+  LAST_ENTRY_LAST_LINE = "Seat 9: ATTACCA folded before Flop (didn't bet)"
+  
+  TABLE_OF_STARTING_INDICES = [0, 1526, 2997, 4594, 5939, 7650, 9174, 10740, 12137, 13446, 14589, 15823]
+
+  before do
+    @psfile = PokerstarsFile.open(File.dirname(__FILE__) + '/file_many_hands.txt')
+    @expanded_path = File.expand_path(File.dirname(__FILE__) + '/file_many_hands.txt')
+  end
+  
+  it "should be open" do
+    @psfile.should_not be_closed
+  end
+  
+  it "should not be eof" do
+    @psfile.should_not be_eof
+  end
+  
+  it "should read all the hands in the test file" do
+    @psfile.should have(NUMBER_OF_ENTRIES).entries
+  end
+  
+  it "should collect entries with all the proper information" do
+    list = TABLE_OF_STARTING_INDICES.clone
+    @psfile.entries.each do |handrecord|
+      handrecord.source.should == @expanded_path
+      handrecord.position.should == list.shift
+    end
+  end
+  
+  it "should be able to access records through valid positions" do
+    TABLE_OF_STARTING_INDICES.each do |index|
+      @entry = @psfile.first(index)
+    end
+  end
+  
+  it "should complain when attempting to reach records through invalid positions" do
+    TABLE_OF_STARTING_INDICES.each do |index|
+      lambda {@psfile.pos=each+1}.should raise_error
+    end
+  end
+  
+  it "should have a first entry having the correct lines, addressable through #first" do
+    @psfile.entries #run through the file to see if it resets properly
+    @entry = @psfile.first(TABLE_OF_STARTING_INDICES.first)
+    @entry.should have(FIRST_ENTRY_NUMBER_OF_LINES).lines
+    @entry.lines.first.should == FIRST_ENTRY_FIRST_LINE
+    @entry.lines.last.should == FIRST_ENTRY_LAST_LINE
+    @psfile.should be_closed
+  end
+  
+  it "should have a first entry having the correct lines, addresable through #entries" do
+    @entries = @psfile.entries.first
+    @entries.should have(FIRST_ENTRY_NUMBER_OF_LINES).lines
+    @entries.lines.first.should == FIRST_ENTRY_FIRST_LINE
+    @entries.lines.last.should == FIRST_ENTRY_LAST_LINE
+  end
+  
+  it "should have a last entry having the correct lines, addressable through #first" do
+    @entry = @psfile.first(TABLE_OF_STARTING_INDICES.last)
+    @entry.should have(LAST_ENTRY_NUMBER_OF_LINES).lines
+    @entry.lines.first.should == LAST_ENTRY_FIRST_LINE
+    @entry.lines.last.should == LAST_ENTRY_LAST_LINE
+  end
+
+  it "should have a last entry having the correct lines, addressable through #entries" do
+    @entries = @psfile.entries.last
+    @entries.should have(LAST_ENTRY_NUMBER_OF_LINES).lines
+    @entries.lines.first.should == LAST_ENTRY_FIRST_LINE
+    @entries.lines.last.should == LAST_ENTRY_LAST_LINE
+  end
+  
+  it "should be at the end of file after reading all the entries" do
+    @psfile.entries
+    @psfile.should be_eof
+  end
+end