2018-03-04 10:09:32 -05:00
|
|
|
require_relative '../../spec_helper'
|
2017-05-07 08:04:49 -04:00
|
|
|
require 'set'
|
|
|
|
|
|
|
|
describe "Set#divide" do
|
|
|
|
it "divides self into a set of subsets based on the blocks return values" do
|
|
|
|
set = Set["one", "two", "three", "four", "five"].divide { |x| x.length }
|
|
|
|
set.map { |x| x.to_a.sort }.sort.should == [["five", "four"], ["one", "two"], ["three"]]
|
|
|
|
end
|
|
|
|
|
|
|
|
it "yields each Object to the block" do
|
|
|
|
ret = []
|
|
|
|
Set["one", "two", "three", "four", "five"].divide { |x| ret << x }
|
|
|
|
ret.sort.should == ["five", "four", "one", "three", "two"]
|
|
|
|
end
|
|
|
|
|
|
|
|
# BUG: Does not raise a LocalJumpError, but a NoMethodError
|
|
|
|
#
|
|
|
|
# it "raises a LocalJumpError when not passed a block" do
|
|
|
|
# lambda { Set[1].divide }.should raise_error(LocalJumpError)
|
|
|
|
# end
|
|
|
|
end
|
|
|
|
|
|
|
|
describe "Set#divide when passed a block with an arity of 2" do
|
|
|
|
it "divides self into a set of subsets based on the blocks return values" do
|
|
|
|
set = Set[1, 3, 4, 6, 9, 10, 11].divide { |x, y| (x - y).abs == 1 }
|
|
|
|
set.map{ |x| x.to_a.sort }.sort.should == [[1], [3, 4], [6], [9, 10, 11]]
|
|
|
|
end
|
|
|
|
|
|
|
|
it "yields each two Object to the block" do
|
|
|
|
ret = []
|
|
|
|
Set[1, 2].divide { |x, y| ret << [x, y] }
|
|
|
|
ret.sort.should == [[1, 1], [1, 2], [2, 1], [2, 2]]
|
|
|
|
end
|
|
|
|
end
|