Sha256: 89d33dad2c012f32357032f684bdb4b62bf351d8517ae0cf218c93195e891c06
Contents?: true
Size: 1010 Bytes
Versions: 17
Compression:
Stored size: 1010 Bytes
Contents
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
Version data entries
17 entries across 17 versions & 6 rubygems