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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
module Test
module Unit
class DataSets
def initialize
@variables = []
@procs = []
@value_sets = []
end
def add(data_set, options=nil)
options ||= {}
if data_set.respond_to?(:call)
@procs << [data_set, options]
elsif data_set.is_a?(Array)
@variables << [data_set, options]
else
@value_sets << [data_set, options]
end
end
def <<(data_set)
add(data_set)
end
def keep
new_data_sets = self.class.new
all_data_sets = Enumerator.new do |yielder|
block = lambda do |(data_set, options)|
yielder << [data_set, options]
end
@procs.each(&block)
@variables.each(&block)
@value_sets.each(&block)
end
all_data_sets.each do |data_set, options|
next if options.nil?
next unless options[:keep]
new_data_sets.add(data_set, options)
end
new_data_sets
end
def each
variables = @variables
value_sets = @value_sets
@procs.each do |proc, options|
data_set = proc.call
case data_set
when Array
variables += [[data_set, options]]
else
value_sets += [[data_set, options]]
end
end
value_sets.each do |values, _options|
values.each do |label, data|
yield(label, data)
end
end
each_pattern(variables) do |label, data|
yield(label, data)
end
end
def ==(other)
@variables == other.instance_variable_get(:@variables) and
@procs == other.instance_variable_get(:@procs) and
@value_sets == other.instance_variable_get(:@value_sets)
end
def eql?(other)
self == other
end
def hash
[@variables, @procs, @value_sets].hash
end
private
def each_pattern(variables)
grouped_variables = variables.group_by do |_, options|
options[:group]
end
grouped_variables.each do |group, group_variables|
each_raw_pattern(group_variables) do |cell|
label = String.new
label << "group: #{group.inspect}" unless group.nil?
data = {}
cell.each do |variable, pattern|
label << ", " unless label.empty?
label << "#{variable}: #{pattern.inspect}"
data[variable] = pattern
end
yield(label, data)
end
end
end
def each_raw_pattern(variables, &block)
return if variables.empty?
sorted_variables = variables.sort_by do |(variable, _), _|
variable
end
all_patterns = sorted_variables.collect do |(variable, patterns), _|
patterns.collect do |pattern|
[variable, pattern]
end
end
all_patterns[0].product(*all_patterns[1..-1], &block)
end
end
end
end
|
