Move db verification into tests

2 files changed, 5 insertions, 10 deletions

diff --git a/lib/linguist/classifier.rb b/lib/linguist/classifier.rb
index 843ffa3..a7f3a16 100644
--- a/lib/linguist/classifier.rb
+++ b/lib/linguist/classifier.rb
@@ -57,15 +57,6 @@ module Linguist
       nil
     end
 
-    # Public: Verify internal counts are consistent.
-    #
-    # Returns Boolean.
-    def verify
-      @languages.inject(0) { |n, (_, c)| n += c } == @languages_total &&
-        @language_tokens.inject(0) { |n, (_, c)| n += c } == @tokens_total &&
-        @tokens.inject(0) { |n, (_, ts)| n += ts.inject(0) { |m, (_, c)| m += c } } == @tokens_total
-    end
-
     # Public: Prune infrequent tokens.
     #
     # Returns receiver Classifier instance.
diff --git a/test/test_classifier.rb b/test/test_classifier.rb
index 0571f85..b0923fc 100644
--- a/test/test_classifier.rb
+++ b/test/test_classifier.rb
@@ -64,7 +64,11 @@ class TestClassifier < Test::Unit::TestCase
   end
 
   def test_verify
-    assert Classifier.instance.verify
+    data = Classifier.instance.to_hash
+
+    assert_equal data['languages_total'], data['languages'].inject(0) { |n, (_, c)| n += c }
+    assert_equal data['tokens_total'], data['language_tokens'].inject(0) { |n, (_, c)| n += c }
+    assert_equal data['tokens_total'], data['tokens'].inject(0) { |n, (_, ts)| n += ts.inject(0) { |m, (_, c)| m += c } }
   end
 
   def test_gc