lib/fiscal_year/quarter.rb in fiscal_year-0.2.0 vs lib/fiscal_year/quarter.rb in fiscal_year-0.3.0

@@ -24,11 +24,14 @@
       def fourth?(month)
         FiscalYear.quarters.fourth.include?(month)
       end
 
       def months(month)
-        FiscalYear.quarters.find { |a| a.include?(month) }
+        months = FiscalYear.quarters.find { |a| a.include?(month) }
+        raise ::StandardError if months.nil?
+
+        months
       end
 
       def range_by(date)
         year = date.year
         this_quater = months(date.month)
@@ -36,10 +39,12 @@
 
         Date.parse("#{year}/#{this_quater.first}/01")..Date.parse("#{last_year}/#{this_quater.last}/01").end_of_month
       end
 
       def quater_num(month)
-        (FiscalYear.quarters.rindex(months(month)) + 1)
+        rindex = FiscalYear.quarters.rindex(months(month))
+
+        rindex.nil? ? 0 : (rindex + 1)
       end
 
       def cross_year_in_quarter?(quarter)
         FiscalYear.cross_year? && quarter.any? { |month| month == 12 }
       end