depth_chart.rb in sportradar-api-0.1.36

- old
+ new

@@ -1,21 +1,30 @@
 module Sportradar
   module Api
     class Nfl::DepthChart < Data
+      include Enumerable
       attr_accessor :response, :chart
 
       def initialize(data)
         generate_chart(data)
       end
 
       def team(number)
         teams[number]
       end
+      def each
+        populate_teams
+        teams.each { |team| yield team }
+      end
 
       private
 
       def teams
         @teams ||= Hash.new { |hash, number| hash[number] = generate_team(number) }
+      end
+
+      def populate_teams
+        (1..3).each(&method(:team)) if teams.empty?
       end
 
       def generate_team(number)
         @chart.each_with_object({}) do |(pos_name, groups), memo|
           memo[pos_name] = groups[number.to_s]