lib/test/unit/hash_test.rb in omnicat-0.1.3 vs lib/test/unit/hash_test.rb in omnicat-0.2.0

@@ -1,10 +1,12 @@
 require File.expand_path(File.join(File.dirname(__FILE__), '..', 'test_helper'))
 
 class TestHash < Test::Unit::TestCase
   def test_to_hash
-    categories_hash = { "pos" => { doc_count: 0, prior: 0.0, tokens: {}, token_count: 0 } }
+    categories_hash = { 
+      "pos" => { doc_count: 0, docs: {}, tokens: {}, token_count: 0 } 
+    }
     categories = OmniCat::Hash.new
-    categories["pos"] = OmniCat::Classifiers::BayesInternals::Category.new(categories_hash["pos"])
+    categories["pos"] = OmniCat::Classifiers::StrategyInternals::Category.new(categories_hash["pos"])
     assert_equal(categories_hash, categories.to_hash)
   end
 end