# coding:utf-8
$KCODE = 'utf8'

require File.dirname(__FILE__) + '/../test_helper'
class BayesianTest < Test::Unit::TestCase
	def setup
		@classifier = Classifier::Bayes.new :categories => ['Interesting', 'Uninteresting']
	end
	
	def test_good_training
		assert_nothing_raised { @classifier.train_interesting "love" }
	end

	def test_bad_training
		assert_raise(StandardError) { @classifier.train_no_category "words" }
	end
	
	def test_bad_method
		assert_raise(NoMethodError) { @classifier.forget_everything_you_know "" }
	end
	
	def test_categories
		assert_equal ['Interesting', 'Uninteresting'].sort, @classifier.categories.sort
	end

	def test_add_category
		@classifier.add_category 'Test'
		assert_equal ['Test', 'Interesting', 'Uninteresting'].sort, @classifier.categories.sort
	end

	def test_classification
		@classifier.train_interesting "here are some good words. I hope you love them"
		@classifier.train_uninteresting "here are some bad words, I hate you"
		assert_equal 'Uninteresting', @classifier.classify("I hate bad words and you")
	end
	
	def test_ru_classification
	  c = Classifier::Bayes.new :categories => ['Interesting', 'Uninteresting'], :language => "ru"
	  c.train_interesting "вот несколько хороших слов. Я надеюсь вам они понравились"
	  c.train_uninteresting "вот несколько плохих слов. Я тебя ненавижу"
	  assert_equal 'Uninteresting', c.classify("Я ненавижу плохие слова и тебя")
  end
  
  def test_case_insensitive
	  c = Classifier::Bayes.new :categories => [:good, :bad], :language => "ru"
	  c.train_good "Хорошо"
	  c.train_bad "Плохо"
	  
	  assert_equal c.classifications("ХОРОШО"), c.classifications("хорошо")
	  assert_equal c.classifications("плОХО"), c.classifications("плохо")
  end
end