class Test::Unit::DataSets

# File lib/test/unit/data-sets.rb, line 4
def initialize
  @variables = []
  @procs = []
  @value_sets = []
end

# File lib/test/unit/data-sets.rb, line 21
def <<(data_set)
  add(data_set)
end

# File lib/test/unit/data-sets.rb, line 67
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

# File lib/test/unit/data-sets.rb, line 10
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

# File lib/test/unit/data-sets.rb, line 43
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

# File lib/test/unit/data-sets.rb, line 73
def eql?(other)
  self == other
end

# File lib/test/unit/data-sets.rb, line 77
def hash
  [@variables, @procs, @value_sets].hash
end

# File lib/test/unit/data-sets.rb, line 25
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

# File lib/test/unit/data-sets.rb, line 82
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

# File lib/test/unit/data-sets.rb, line 101
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