minor refactor in order to make the count method smaller

giusepped · Nov 24, 2015 · 31a961f · 31a961f
1 parent 9314987
commit 31a961f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/counter.rb b/counter.rb
@@ -1,8 +1,7 @@
 require_relative './lib/palindrome_counter.rb'
 
-counter = PalindromeCounter.new
-
-Dir.glob('test_files/*.txt') do |test_file|
+Dir.glob('**/*.txt') do |test_file|
+  counter = PalindromeCounter.new
   range_numbers = File.foreach(test_file).first(2)
   start = range_numbers[0].chomp
   finish = range_numbers[1]
@@ -11,4 +10,4 @@
   puts "Range: #{start}..#{finish}"
   puts "Palindromes: #{palindrome_count}"
   puts "\n"
-end
+end
diff --git a/lib/palindrome_counter.rb b/lib/palindrome_counter.rb
@@ -1,11 +1,10 @@
 class PalindromeCounter
 
   def initialize
-    @counter
+    @counter = 0
   end
 
   def count(start, finish)
-    @counter = 0
     (start..finish).each do |num|
       @counter += 1 if check_palindrome_recursive(num.to_s)
     end