summaryrefslogtreecommitdiff
path: root/test/-ext-/arith_seq/test_arith_seq_extract.rb
blob: d64fe6a7b5134c7fe5ba8b650da93f946ebcac73 (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
# frozen_string_literal: false
require 'test/unit'

class Test_ArithSeq < Test::Unit::TestCase
  def test_extract_with_arith_seq
    assert_separately([], <<-"end;") #do
      require '-test-/arith_seq/extract'

      b, e, s, f, r = Enumerator::ArithmeticSequence.__extract__(1.step(10, 2))
      assert_equal([1, 10, 2, 0, 1], [b, e, s, f, r])

      b, e, s, f, r = Enumerator::ArithmeticSequence.__extract__((1..10) % 2)
      assert_equal([1, 10, 2, 0, 1], [b, e, s, f, r])

      b, e, s, f, r = Enumerator::ArithmeticSequence.__extract__((1...10) % 2)
      assert_equal([1, 10, 2, 1, 1], [b, e, s, f, r])
    end;
  end

  def test_extract_with_range
    assert_separately([], <<-"end;") #do
      require '-test-/arith_seq/extract'

      b, e, s, f, r = Enumerator::ArithmeticSequence.__extract__(1..10)
      assert_equal([1, 10, 1, 0, 1], [b, e, s, f, r])

      b, e, s, f, r = Enumerator::ArithmeticSequence.__extract__(1...10)
      assert_equal([1, 10, 1, 1, 1],  [b, e, s, f, r])
    end;
  end

  def test_extract_with_others
    assert_separately([], <<-"end;") #do
      require '-test-/arith_seq/extract'

      b, e, s, f, r = Enumerator::ArithmeticSequence.__extract__(nil)
      assert_equal([nil, nil, nil, nil, 0], [b, e, s,  f, r])
    end;
  end
end