sapor 0.3.3

data/spec/unit/belgium_brussels_spec.rb

@@ -0,0 +1,58 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2016 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+require 'spec_helper'
+
+describe Sapor::BelgiumBrussels, '#area_code' do
+  it 'returns BE[BRU] as the area code' do
+    expect(Sapor::BelgiumBrussels.instance.area_code).to eq('BE[BRU]')
+  end
+end
+
+describe Sapor::BelgiumBrussels, '#no_of_seats' do
+  it 'returns 15 as the number of seats' do
+    expect(Sapor::BelgiumBrussels.instance.no_of_seats).to eq(15)
+  end
+end
+
+describe Sapor::BelgiumBrussels, '#population_size' do
+  it 'returns a population size of 501,459' do
+    expect(Sapor::BelgiumBrussels.instance.population_size).to eq(501_459)
+  end
+end
+
+describe Sapor::BelgiumBrussels, '#seats' do
+  it 'calculates the number of seats for the election of 2014' do
+    belgium_brussels = Sapor::BelgiumBrussels.instance
+    results = belgium_brussels.overall_election_results_of_2019
+    seats = belgium_brussels.seats(results)
+    expect(seats['Ecolo']).to eq(4)
+    expect(seats['Parti Socialiste']).to eq(3)
+    expect(seats['Mouvement Réformateur']).to eq(3)
+    expect(seats['Parti du Travail de Belgique']).to eq(2)
+    expect(seats['DéFI']).to eq(2)
+    expect(seats['Centre démocrate humaniste']).to eq(1)
+    expect(seats['Nieuw-Vlaamse Alliantie']).to eq(0)
+    expect(seats['Listes Destexhe']).to eq(0)
+    expect(seats['Open Vlaamse Liberalen en Democraten']).to eq(0)
+    expect(seats['Parti Populaire']).to eq(0)
+    expect(seats['Vlaams Belang']).to eq(0)
+    expect(seats['Christen-Democratisch en Vlaams']).to eq(0)
+  end
+end

data/spec/unit/belgium_flanders_spec.rb

@@ -0,0 +1,62 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2016 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+require 'spec_helper'
+
+describe Sapor::BelgiumFlanders, '#area_code' do
+  it 'returns BE[VLG] as the area code' do
+    expect(Sapor::BelgiumFlanders.instance.area_code).to eq('BE[VLG]')
+  end
+end
+
+describe Sapor::BelgiumFlanders, '#no_of_seats' do
+  it 'returns 87 as the number of seats' do
+    expect(Sapor::BelgiumFlanders.instance.no_of_seats).to eq(87)
+  end
+end
+
+describe Sapor::BelgiumFlanders, '#population_size' do
+  it 'returns a population size of 4,206,305' do
+    expect(Sapor::BelgiumFlanders.instance.population_size).to eq(4_206_305)
+  end
+end
+
+describe Sapor::BelgiumFlanders, '#seats' do
+  it 'calculates the number of seats for the election of 2019' do
+    belgium_flanders = Sapor::BelgiumFlanders.instance
+    results = belgium_flanders.overall_election_results_of_2019
+    seats = belgium_flanders.seats(results)
+    expect(seats['Nieuw-Vlaamse Alliantie']).to eq(25)
+    expect(seats['Vlaams Belang']).to eq(18)
+    expect(seats['Open Vlaamse Liberalen en Democraten']).to eq(12)
+    expect(seats['Christen-Democratisch en Vlaams']).to eq(12)
+    expect(seats['Socialistische Partij Anders']).to eq(9)
+    expect(seats['Groen']).to eq(8)
+    expect(seats['Partij van de Arbeid van België']).to eq(3)
+    expect(seats['Belgische Unie – Union Belge']).to eq(0)
+    expect(seats['de coöperatie']).to eq(0)
+    expect(seats['DéFI']).to eq(0)
+    expect(seats['Democratisch-Solidair Appèl']).to eq(0)
+    expect(seats['DierAnimal']).to eq(0)
+    expect(seats['Partij Vrede en Solidariteit']).to eq(0)
+    expect(seats['Piratenpartij']).to eq(0)
+    expect(seats['PRO']).to eq(0)
+    expect(seats['Volt']).to eq(0)
+  end
+end

data/spec/unit/belgium_spec.rb

@@ -0,0 +1,26 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2016 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+require 'spec_helper'
+
+describe Sapor::Belgium, '#threshold' do
+  it 'returns a threshold of 5%' do
+    expect(Sapor::Belgium.instance.threshold).to eq(0.05)
+  end
+end

data/spec/unit/belgium_wallonia_spec.rb

@@ -0,0 +1,65 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2016 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+require 'spec_helper'
+
+describe Sapor::BelgiumWallonia, '#area_code' do
+  it 'returns BE[WAL] as the area code' do
+    expect(Sapor::BelgiumWallonia.instance.area_code).to eq('BE[WAL]')
+  end
+end
+
+describe Sapor::BelgiumWallonia, '#no_of_seats' do
+  it 'returns 48 as the number of seats' do
+    expect(Sapor::BelgiumWallonia.instance.no_of_seats).to eq(48)
+  end
+end
+
+describe Sapor::BelgiumWallonia, '#population_size' do
+  it 'returns a population size of 2,072,774' do
+    expect(Sapor::BelgiumWallonia.instance.population_size).to eq(2_072_774)
+  end
+end
+
+describe Sapor::BelgiumWallonia, '#seats' do
+  it 'calculates the number of seats for the election of 2019' do
+    belgium_wallonia = Sapor::BelgiumWallonia.instance
+    results = belgium_wallonia.overall_election_results_of_2019
+    seats = belgium_wallonia.seats(results)
+    expect(seats['Parti Socialiste']).to eq(17)
+    expect(seats['Mouvement Réformateur']).to eq(11)
+    expect(seats['Ecolo']).to eq(9)
+    expect(seats['Parti du Travail de Belgique']).to eq(7)
+    expect(seats['Centre démocrate humaniste']).to eq(4)
+    expect(seats['DéFI']).to eq(0)
+    expect(seats['AGIR']).to eq(0)
+    expect(seats['Collectif Citoyen']).to eq(0)
+    expect(seats['DierAnimal']).to eq(0)
+    expect(seats['La Droite']).to eq(0)
+    expect(seats['Les Belges d’Abord']).to eq(0)
+    expect(seats['Listes Destexhe']).to eq(0)
+    expect(seats['Lutte Ouvrière']).to eq(0)
+    expect(seats['NATION']).to eq(0)
+    expect(seats['Parti Communiste de Belgique']).to eq(0)
+    expect(seats['Parti Populaire']).to eq(0)
+    expect(seats['Turquoise']).to eq(0)
+    expect(seats['Vlaams Belang']).to eq(0)
+    expect(seats['Wallonie Insoumise']).to eq(0)
+  end
+end

data/spec/unit/binomials_cache_spec.rb

@@ -0,0 +1,34 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2016 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+require 'spec_helper'
+
+describe Sapor::BinomialsCache, '#binomial' do
+  it 'returns the correct binomial' do
+    actual = Sapor::BinomialsCache.binomial(10, 20)
+    expected = 10.large_float_binomial_by_product_of_divisions(20)
+    expect(actual).to eq(expected)
+  end
+
+  it 'returns the same binomial when asked a second time' do
+    first = Sapor::BinomialsCache.binomial(10, 21)
+    second = Sapor::BinomialsCache.binomial(10, 21)
+    expect(second).to eq(first)
+  end
+end

data/spec/unit/catalonia_spec.rb

@@ -0,0 +1,74 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2016 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+require 'spec_helper'
+
+describe Sapor::Catalonia, '#area_code' do
+  it 'returns ES-CT as the area code' do
+    expect(Sapor::Catalonia.instance.area_code).to eq('ES-CT')
+  end
+end
+
+describe Sapor::Catalonia, '#coalitions' do
+  it 'returns JS and JS+CUP as the coalitions to track' do
+    expect(Sapor::Catalonia.instance.coalitions).to include(['Junts pel Sí'])
+    expect(Sapor::Catalonia.instance.coalitions).to \
+      include(['Junts pel Sí', 'Candidatura d’Unitat Popular'])
+    expect(Sapor::Catalonia.instance.coalitions).to \
+      include(['Ciutadans–Partido de la Ciudadanía',
+               'Partit dels Socialistes de Catalunya (PSC-PSOE)',
+               'Catalunya en Comú', 'Partit Popular'])
+  end
+end
+
+describe Sapor::Catalonia, '#no_of_seats' do
+  it 'returns 135 as the number of seats' do
+    expect(Sapor::Catalonia.instance.no_of_seats).to eq(135)
+  end
+end
+
+describe Sapor::Catalonia, '#population_size' do
+  it 'returns a population size of 5,510,853' do
+    expect(Sapor::Catalonia.instance.population_size).to eq(5_510_853)
+  end
+end
+
+describe Sapor::Catalonia, '#seats' do
+  it 'calculates the number of seats for the election of 2015 correctly' do
+    catalonia = Sapor::Catalonia.instance
+    results = catalonia.overall_election_results_of_2015
+    seats = catalonia.seats(results)
+    expect(seats['Junts pel Sí']).to eq(62)
+    expect(seats['Ciutadans–Partido de la Ciudadanía']).to eq(25)
+    expect(seats['Partit dels Socialistes de Catalunya (PSC-PSOE)']).to eq(16)
+    expect(seats['Catalunya en Comú']).to eq(11)
+    expect(seats['Partit Popular']).to eq(11)
+    expect(seats['Candidatura d’Unitat Popular']).to eq(10)
+    expect(seats['Unió Democràtica de Catalunya']).to eq(0)
+    expect(seats['Partit Animalista contra el Maltractament Animal']).to eq(0)
+    expect(seats['Pirates de Catalunya–Per decidir-ho tot']).to eq(0)
+    expect(seats['Guanyem Catalunya']).to eq(0)
+  end
+end
+
+describe Sapor::Catalonia, '#threshold' do
+  it 'returns a threshold of 3%' do
+    expect(Sapor::Catalonia.instance.threshold).to eq(0.03)
+  end
+end

data/spec/unit/combinations_distribution_spec.rb

@@ -0,0 +1,241 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2016 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+require 'spec_helper'
+
+describe Sapor::CombinationsDistribution, '#new' do
+  it 'creates an empty distribution' do
+    distribution = Sapor::CombinationsDistribution.new
+    expect(distribution).to be_empty
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#[]=' do
+  it 'sets a value-combinations pair' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[1] = 2
+    expect(distribution[1]).to eq(2)
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#+' do
+  it 'adds two combiation distributions together' do
+    a = Sapor::CombinationsDistribution.new
+    a[10] = 2
+    a[21] = 3
+    a[23] = 11
+    b = Sapor::CombinationsDistribution.new
+    b[10] = 5
+    b[21] = 7
+    b[29] = 13
+    sum = Sapor::CombinationsDistribution.new
+    sum[10] = 7
+    sum[21] = 10
+    sum[23] = 11
+    sum[29] = 13
+    expect((a + b)[10]).to eq(sum[10])
+    expect((a + b)[21]).to eq(sum[21])
+    expect((a + b)[23]).to eq(sum[23])
+    expect((a + b)[29]).to eq(sum[29])
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#confidence_interval' do
+  it 'returns the single value as boundaries (default)' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[5] = 2
+    expect(distribution.confidence_interval(0.95)).to eq([5, 5])
+  end
+
+  it 'returns the full interval as boundaries if single value (extensive)' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[5] = 2
+    expect(distribution.confidence_interval(0.95, 10)).to eq([0, 10])
+  end
+
+  it 'returns the boundaries (default)' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[3] = 2
+    distribution[7] = 2
+    expect(distribution.confidence_interval(0.95)).to eq([3, 7])
+  end
+
+  it 'returns the full interval as boundaries (extensive)' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[3] = 2
+    distribution[7] = 2
+    expect(distribution.confidence_interval(0.95, 10)).to eq([0, 10])
+  end
+
+  it 'returns the full interval if level too high' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[5] = 1
+    distribution[10] = 98
+    distribution[15] = 1
+    expect(distribution.confidence_interval(0.99)).to eq([5, 15])
+  end
+
+  it 'returns the values in between as boundaries (even)' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[5] = 1
+    distribution[10] = 98
+    distribution[15] = 1
+    expect(distribution.confidence_interval(0.95)).to eq([8, 12])
+  end
+
+  it 'returns the values in between as boundaries (odd)' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[6] = 1.to_lf
+    distribution[10] = 98.to_lf
+    distribution[14] = 1.to_lf
+    expect(distribution.confidence_interval(0.95)).to eq([8, 12])
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#confidence_interval_values' do
+  it 'extracts the single value as confidence interval value' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[5] = 2.to_lf
+    expect(distribution.confidence_interval_values(0.95)).to eq([5])
+  end
+
+  it 'extracts all values as confidence intervalue values' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[3] = 2.to_lf
+    distribution[7] = 2.to_lf
+    expect(distribution.confidence_interval_values(0.95).sort).to eq([3, 7])
+  end
+
+  it 'extracts the middle values as confidence interval values' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[5] = 1.to_lf
+    distribution[10] = 98.to_lf
+    distribution[15] = 1.to_lf
+    expect(distribution.confidence_interval_values(0.95)).to eq([10])
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#empty?' do
+  it 'returns true if no value-combinations pair has been added' do
+    distribution = Sapor::CombinationsDistribution.new
+    expect(distribution.empty?).to be true
+  end
+
+  it 'returns false if a value-combinations pair has been added' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[1] = 2.to_lf
+    expect(distribution.empty?).to be false
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#most_probable_rounded_fraction' do
+  it "returns the median rounded fraction if there's more than one with" \
+     ' maximal probability' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[500] = 2.to_lf
+    expect(distribution.most_probable_rounded_fraction(1000)).to eq(0.5)
+  end
+
+  it 'returns the rounded fraction with the maximal probability' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[199] = 2.to_lf
+    distribution[200] = 3.to_lf
+    distribution[201] = 2.to_lf
+    expect(distribution.most_probable_rounded_fraction(1000)).to eq(0.2)
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#most_probable_value' do
+  it "returns the single value if there's only one value" do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[1] = 2
+    expect(distribution.most_probable_value).to eq(1)
+  end
+
+  it 'returns the first of two values if that has the greatest number ' \
+     'of combinations' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[1] = 2
+    distribution[2] = 1
+    expect(distribution.most_probable_value).to eq(1)
+  end
+
+  it 'returns the second of two values if that has the greatest number ' \
+     'of combinations' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[1] = 1
+    distribution[2] = 2
+    expect(distribution.most_probable_value).to eq(2)
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#size' do
+  it 'returns 0 for an empty distribution' do
+    distribution = Sapor::CombinationsDistribution.new
+    expect(distribution.size).to eq(0)
+  end
+
+  it 'returns 1 when a value-combinations pair has been added' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[1] = 2
+    expect(distribution.size).to eq(1)
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#threshold_probability' do
+  it 'returns 1 if the single value is above the threshold' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[5] = 2.to_lf
+    expect(distribution.threshold_probability(0.1, 10)).to eq(1)
+  end
+
+  it 'returns 0 if the single value is below the threshold' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[5] = 2.to_lf
+    expect(distribution.threshold_probability(0.6, 10)).to eq(0)
+  end
+
+  it 'returns 0.8 if the threshold is between combinations 2 and 8' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[2] = 2.to_lf
+    distribution[7] = 8.to_lf
+    expect(distribution.threshold_probability(0.5, 10)).to eq(0.8)
+  end
+
+  it 'includes the probability at the threshold' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[2] = 2.to_lf
+    distribution[5] = 5.to_lf
+    distribution[7] = 3.to_lf
+    expect(distribution.threshold_probability(0.5, 10)).to eq(0.8)
+  end
+end
+
+describe Sapor::CombinationsDistribution, '#values' do
+  it 'returns an empty array for an empty distribution' do
+    distribution = Sapor::CombinationsDistribution.new
+    expect(distribution.values).to eq([])
+  end
+
+  it 'returns 1 when a value-combinations pair has been added' do
+    distribution = Sapor::CombinationsDistribution.new
+    distribution[1] = 2
+    expect(distribution.values).to eq([1])
+  end
+end