diff --git a/ChangeLog b/ChangeLog index 5b47e93883..8a85afc124 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,9 @@ +Tue Jul 30 18:52:27 2013 Akinori MUSHA + + * lib/set.rb (Set#intersect?, Set#disjoint?): Add new methods for + testing if two sets have any element in common. + [ruby-core:45641] [Feature #6588] Based on the code by marcandre. + Tue Jul 30 17:16:15 2013 Nobuyoshi Nakada * sprintf.c (ruby__sfvextra): add QUOTE flag to escape unprintable diff --git a/NEWS b/NEWS index 31a856befa..ee83b1b4c6 100644 --- a/NEWS +++ b/NEWS @@ -141,6 +141,11 @@ with all sufficient information, see the ChangeLog file. * REXML::Text#<< supports method chain like 'text << "XXX" << "YYY"'. * REXML::Text#<< supports not "raw" mode. +* Set + * New methods: + * Set#intersect? + * Set#disjoint? + === Stdlib compatibility issues (excluding feature bug fixes) * Set diff --git a/lib/set.rb b/lib/set.rb index 47fcc3efc7..aec22ef4fe 100644 --- a/lib/set.rb +++ b/lib/set.rb @@ -231,6 +231,23 @@ class Set end alias < proper_subset? + # Returns true if the set and the given set have at least one + # element in common. + def intersect?(set) + set.is_a?(Set) or raise ArgumentError, "value must be a set" + if size < set.size + any? { |o| set.include?(o) } + else + set.any? { |o| include?(o) } + end + end + + # Returns true if the set and the given set have no element in + # common. This method is the opposite of +intersect?+. + def disjoint?(set) + !intersect?(set) + end + # Calls the given block once for each element in the set, passing # the element as parameter. Returns an enumerator if no block is # given. diff --git a/test/test_set.rb b/test/test_set.rb index cbf4eb0234..1f6f2436f5 100644 --- a/test/test_set.rb +++ b/test/test_set.rb @@ -297,6 +297,46 @@ class TC_Set < Test::Unit::TestCase assert_equal(false, Set[].proper_subset?(Set[])) end + def assert_intersect(expected, set, other) + case expected + when true + assert_send([set, :intersect?, other]) + assert_send([other, :intersect?, set]) + assert_not_send([set, :disjoint?, other]) + assert_not_send([other, :disjoint?, set]) + when false + assert_not_send([set, :intersect?, other]) + assert_not_send([other, :intersect?, set]) + assert_send([set, :disjoint?, other]) + assert_send([other, :disjoint?, set]) + when Class + assert_raises(expected) { + set.intersect?(other) + } + assert_raises(expected) { + set.disjoint?(other) + } + else + raise ArgumentError, "%s: unsupported expected value: %s" % [__method__, expected.inspect] + end + end + + def test_intersect? + set = Set[3,4,5] + + assert_intersect(ArgumentError, set, 3) + assert_intersect(ArgumentError, set, [2,4,6]) + + assert_intersect(true, set, Set[2,4]) + assert_intersect(true, set, Set[5,6,7]) + assert_intersect(true, set, Set[1,2,6,8,4]) + + assert_intersect(false, set, Set[]) + assert_intersect(false, set, Set[0,2]) + assert_intersect(false, set, Set[0,2,6]) + assert_intersect(false, set, Set[0,2,6,8,10]) + end + def test_each ary = [1,3,5,7,10,20] set = Set.new(ary)