lib/rubocop/cop/style/sample.rb in rubocop-0.68.0 vs lib/rubocop/cop/style/sample.rb in rubocop-0.68.1

@@ -75,13 +75,12 @@
             sample_size_for_two_args(*method_args)
           end
         end
 
         def sample_size_for_one_arg(arg)
-          case arg.type
-          when :erange, :irange
+          if arg.range_type?
             range_size(arg)
-          when :int
+          elsif arg.int_type?
             [0, -1].include?(arg.to_a.first) ? nil : :unknown
           else
             :unknown
           end
         end