lib/inch/cli/sparkline_helper.rb in inch-0.2.2 vs lib/inch/cli/sparkline_helper.rb in inch-0.2.3

@@ -1,29 +1,29 @@
 module Inch
   module CLI
     module SparklineHelper
-      def ranges_sparkline(_ranges)
-        ranges = _ranges.reverse
-        list = ranges.map { |r| r.objects.size }
-        sparkline = Sparkr::Sparkline.new(list)
-        sparkline.format do |tick, count, index|
-          t = tick.color(ranges[index].color)
-          index == 0 ? t + ' ' : t
-        end
+      def grade_lists_sparkline(_grade_lists)
+        grade_lists = _grade_lists.reverse
+        list = grade_lists.map { |r| r.objects.size }
+        __sparkline(list, grade_lists)
       end
 
       def grades_sparkline(objects)
         grades = {}
         objects.each do |o|
-          grades[o.grade] ||= 0
-          grades[o.grade] += 1
+          grades[o.grade.to_sym] ||= 0
+          grades[o.grade.to_sym] += 1
         end
-        ranges = Evaluation.new_score_ranges.reverse
-        order = ranges.map(&:grade)
+        grade_lists = Evaluation.new_grade_lists.reverse
+        order = grade_lists.map(&:to_sym)
         list = order.map { |g| grades[g] }
+        __sparkline(list, grade_lists)
+      end
+
+      def __sparkline(list, grade_lists)
         sparkline = Sparkr::Sparkline.new(list)
         sparkline.format do |tick, count, index|
-          t = tick.color(ranges[index].color)
+          t = tick.color(grade_lists[index].color)
           index == 0 ? t + ' ' : t
         end
       end
     end
   end