rares (owner)

Fork Of

Revisions

gist: 90628 Download_button fork
public
Public Clone URL: git://gist.github.com/90628.git
Embed All Files: show embed
snippet.rb #
raw 
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

class FilterableEnumerable
  def initialize(original)
    @original = original
  end
  
  def ==(value)
    @original.select { |v| v == value }
  end
 
  def method_missing(sym, *args, &block)
    @original.select {|v| v.send(sym, *args, &block) }
  end
 
  def respond_to?(sym, include_private = false)
    @original.all? {|v| v.respond_to?(sym, include_private) }
  end
 
end
 
class NegativeFilterableEnumerable
  def initialize(original)
    @original = original
  end
  
  def ==(value)
    @original.select { |v| v != value }
  end
 
  def method_missing(sym, *args, &block)
    @original.select {|v| ! v.send(sym, *args, &block) }
  end
 
  def respond_to?(sym, include_private = false)
    @original.all? {|v| v.respond_to?(sym, include_private) }
  end
 
end
 
module Enumerable
  def filter
    FilterableEnumerable.new(self)
  end
 
  def filter_not
    NegativeFilterableEnumerable.new(self)
  end
end
 
 
require 'test/unit'
class TestFilterableEnumerable < Test::Unit::TestCase
  def setup
    @array = (0..10).to_a
  end
 
  def test_gte
    assert_equal [5, 6, 7, 8, 9, 10], @array.filter >= 5
  end
 
  def test_gt
    assert_equal [6, 7, 8, 9, 10], @array.filter > 5
  end
 
  def test_lte
    assert_equal [0, 1, 2, 3, 4], @array.filter < 5
  end
 
  def test_le
    assert_equal [0, 1, 2, 3, 4, 5], @array.filter <= 5
  end
 
  def test_eq
    assert_equal [5], @array.filter == 5
  end
 
  def test_not_eq
    assert_equal [0, 1, 2, 3, 4, 6, 7, 8, 9, 10], @array.filter_not == 5
  end
 
  def test_method_missing
    assert_equal [0], @array.filter.zero?
  end
 
  def test_not_method_missing
    assert_equal [1, 2, 3, 4, 5, 6, 7, 8, 9, 10], @array.filter_not.zero?
  end
 
  def test_respond_to
    assert @array.filter.respond_to?(:zero?)
  end
 
end