1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/spec/ruby/core/enumerable/grep_spec.rb
Marc-Andre Lafortune d5f0d338c7 Optimize Enumerable#grep{_v}
[Bug #17030]
2020-12-15 12:54:45 -05:00

84 lines
2.5 KiB
Ruby

require_relative '../../spec_helper'
require_relative 'fixtures/classes'
describe "Enumerable#grep" do
before :each do
@a = EnumerableSpecs::EachDefiner.new( 2, 4, 6, 8, 10)
end
it "grep without a block should return an array of all elements === pattern" do
class EnumerableSpecGrep; def ===(obj); obj == '2'; end; end
EnumerableSpecs::Numerous.new('2', 'a', 'nil', '3', false).grep(EnumerableSpecGrep.new).should == ['2']
end
it "grep with a block should return an array of elements === pattern passed through block" do
class EnumerableSpecGrep2; def ===(obj); /^ca/ =~ obj; end; end
EnumerableSpecs::Numerous.new("cat", "coat", "car", "cadr", "cost").grep(EnumerableSpecGrep2.new) { |i| i.upcase }.should == ["CAT", "CAR", "CADR"]
end
it "grep the enumerable (rubycon legacy)" do
EnumerableSpecs::EachDefiner.new().grep(1).should == []
@a.grep(3..7).should == [4,6]
@a.grep(3..7) {|a| a+1}.should == [5,7]
end
it "can use $~ in the block when used with a Regexp" do
ary = ["aba", "aba"]
ary.grep(/a(b)a/) { $1 }.should == ["b", "b"]
end
it "sets $~ in the block" do
"z" =~ /z/ # Reset $~
["abc", "def"].grep(/b/) { |e|
e.should == "abc"
$&.should == "b"
}
# Set by the failed match of "def"
$~.should == nil
end
ruby_version_is ""..."3.0.0" do
it "sets $~ to the last match when given no block" do
"z" =~ /z/ # Reset $~
["abc", "def"].grep(/b/).should == ["abc"]
# Set by the failed match of "def"
$~.should == nil
["abc", "def"].grep(/e/)
$&.should == "e"
end
end
ruby_version_is "3.0.0" do
it "does not set $~ when given no block" do
"z" =~ /z/ # Reset $~
["abc", "def"].grep(/b/).should == ["abc"]
$&.should == "z"
end
end
describe "with a block" do
before :each do
@numerous = EnumerableSpecs::Numerous.new(*(0..9).to_a)
def (@odd_matcher = BasicObject.new).===(obj)
obj.odd?
end
end
it "returns an Array of matched elements that mapped by the block" do
@numerous.grep(@odd_matcher) { |n| n * 2 }.should == [2, 6, 10, 14, 18]
end
it "calls the block with gathered array when yielded with multiple arguments" do
EnumerableSpecs::YieldsMixed2.new.grep(Object){ |e| e }.should == EnumerableSpecs::YieldsMixed2.gathered_yields
end
it "raises an ArgumentError when not given a pattern" do
-> { @numerous.grep { |e| e } }.should raise_error(ArgumentError)
end
end
end