class Minitest::PrideLOL

def initialize io # :nodoc:
  # walk red, green, and blue around a circle separated by equal thirds.
  #
  # To visualize, type this into wolfram-alpha:
  #
  #   plot (3*sin(x)+3), (3*sin(x+2*pi/3)+3), (3*sin(x+4*pi/3)+3)
  @colors = (6 * 7).times.map { |n|
    n *= 1.0 / 3
    r  = (3 * Math.sin(n           ) + 3).to_i
    g  = (3 * Math.sin(n + 4 * PI_3) + 3).to_i
    b  = (3 * Math.sin(n + 2 * PI_3) + 3).to_i
    # Then we take rgb and encode them in a single number using
    # base 6, shifted by 16 for the base 16 ansi colors.
    36 * r + 6 * g + b + 16
  }.rotate(4) # puts "red" first
  super
end

def pride string
  c = @colors[@index % @size]
  @index += 1
  "#{ESC}38;5;#{c}m#{string}#{NND}"
end