summaryrefslogtreecommitdiff
path: root/spec/ruby/library/set/flatten_spec.rb
blob: 4ac83ea82572be67b39667f099e159870274dd3d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
require_relative '../../spec_helper'
require_relative 'fixtures/set_like'
require 'set'

describe "Set#flatten" do
  it "returns a copy of self with each included Set flattened" do
    set = Set[1, 2, Set[3, 4, Set[5, 6, Set[7, 8]]], 9, 10]
    flattened_set = set.flatten

    flattened_set.should_not equal(set)
    flattened_set.should == Set[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
  end

  it "raises an ArgumentError when self is recursive" do
    (set = Set[]) << set
    -> { set.flatten }.should raise_error(ArgumentError)
  end

  context "when Set contains a Set-like object" do
    it "returns a copy of self with each included Set-like object flattened" do
      Set[SetSpecs::SetLike.new([1])].flatten.should == Set[1]
    end
  end
end

describe "Set#flatten!" do
  it "flattens self" do
    set = Set[1, 2, Set[3, 4, Set[5, 6, Set[7, 8]]], 9, 10]
    set.flatten!
    set.should == Set[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
  end

  it "returns self when self was modified" do
    set = Set[1, 2, Set[3, 4]]
    set.flatten!.should equal(set)
  end

  it "returns nil when self was not modified" do
    set = Set[1, 2, 3, 4]
    set.flatten!.should be_nil
  end

  it "raises an ArgumentError when self is recursive" do
    (set = Set[]) << set
    -> { set.flatten! }.should raise_error(ArgumentError)
  end

  context "when Set contains a Set-like object" do
    it "flattens self, including Set-like objects" do
      Set[SetSpecs::SetLike.new([1])].flatten!.should == Set[1]
    end
  end
end