1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
require File.dirname(__FILE__) + '/../test_helper'
class BayesianTest < Test::Unit::TestCase
def setup
@classifier = Classifier::Bayes.new '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
end
|
