Sha256: 67363b840fa4b02156c5bf5eae6ccd9f5ce4ff273e8f575f3272452576a14556

Contents?: true

Size: 1.87 KB

Versions: 36

Compression:

Stored size: 1.87 KB

Contents

require File.expand_path(File.dirname(__FILE__) + '/../../test_helper')

describe ::Inch::Utils::WeightedList do
  before do
    @counts = [4, 8, 8]
  end

  def assert_weighted_list(list, counts, expected)
    weighted_list = ::Inch::Utils::WeightedList.new(list, counts)
    # assert_equal expected.map(&:size).inject(:+), weighted_list.to_a.map(&:size).inject(:+)
    assert_equal expected, weighted_list.to_a, "should be #{expected.map(&:size)}, was #{weighted_list.to_a.map(&:size)}"
  end

  def list_and_expected(counts, expected_counts)
    elements = [:B, :C, :U]
    list = counts.map.with_index do |num, index|
      (1..num).map { |i| :"#{elements[index]}#{i}" }
    end
    expected = expected_counts.map.with_index do |num, index|
      (1..num).map { |i| :"#{elements[index]}#{i}" }
    end
    return list, expected
  end

  it "should work if elements are exact" do
    @list, @expected = list_and_expected([4, 8, 8], [4, 8, 8])
    assert_weighted_list(@list, @counts, @expected)
  end

  it "should work if more than enough elements are present" do
    @list, @expected = list_and_expected([10, 10, 10], [4, 8, 8])
    assert_weighted_list(@list, @counts, @expected)
  end

  it "should work if not enough Bs are present" do
    @list, @expected = list_and_expected([2, 12, 15], [2, 8, 10])
    assert_weighted_list(@list, @counts, @expected)
  end

  it "should work if not enough Cs are present" do
    @list, @expected = list_and_expected([15, 4, 15], [4, 4, 12])
    assert_weighted_list(@list, @counts, @expected)
  end

  it "should work if not enough Us are present" do
    @list, @expected = list_and_expected([15, 15, 4], [4, 12, 4])
    assert_weighted_list(@list, @counts, @expected)
  end

  it "should work if not enough Bs AND Cs and Us are present" do
    @list, @expected = list_and_expected([2, 2, 15], [2, 2, 15])
    assert_weighted_list(@list, @counts, @expected)
  end

end

Version data entries

36 entries across 36 versions & 1 rubygems

Version Path
inch-0.5.0.rc3 test/unit/utils/weighted_list_test.rb
inch-0.5.0.rc2 test/unit/utils/weighted_list_test.rb
inch-0.5.0.rc1 test/unit/utils/weighted_list_test.rb
inch-0.4.6 test/unit/utils/weighted_list_test.rb
inch-0.4.5 test/unit/utils/weighted_list_test.rb
inch-0.4.4 test/unit/utils/weighted_list_test.rb
inch-0.4.4.rc4 test/unit/utils/weighted_list_test.rb
inch-0.4.4.rc3 test/unit/utils/weighted_list_test.rb
inch-0.4.4.rc2 test/unit/utils/weighted_list_test.rb
inch-0.4.4.rc1 test/unit/utils/weighted_list_test.rb
inch-0.4.3 test/unit/utils/weighted_list_test.rb
inch-0.4.3.rc2 test/unit/utils/weighted_list_test.rb
inch-0.4.3.rc1 test/unit/utils/weighted_list_test.rb
inch-0.4.2 test/unit/utils/weighted_list_test.rb
inch-0.4.1 test/unit/utils/weighted_list_test.rb
inch-0.4.0 test/unit/utils/weighted_list_test.rb
inch-0.4.0.rc3 test/unit/utils/weighted_list_test.rb
inch-0.4.0.rc2 test/unit/utils/weighted_list_test.rb
inch-0.4.0.rc1 test/unit/utils/weighted_list_test.rb
inch-0.3.4.rc1 test/unit/utils/weighted_list_test.rb