linguist/test/test_classifier.rb

require 'linguist/classifier'
require 'linguist/language'
require 'linguist/sample'
require 'linguist/tokenizer'

require 'test/unit'

class TestClassifier < Test::Unit::TestCase
  include Linguist

  def fixtures_path
    File.expand_path("../fixtures", __FILE__)
  end

  def fixture(name)
    File.read(File.join(fixtures_path, name))
  end

  def test_classify
    classifier = Classifier.new
    classifier.train Language["Ruby"], fixture("ruby/foo.rb")
    classifier.train Language["Objective-C"], fixture("objective-c/Foo.h")
    classifier.train Language["Objective-C"], fixture("objective-c/Foo.m")

    results = classifier.classify(fixture("objective-c/hello.m"))
    assert_equal Language["Objective-C"], results.first[0]

    tokens  = Tokenizer.new(fixture("objective-c/hello.m")).tokens
    results = classifier.classify(tokens)
    assert_equal Language["Objective-C"], results.first[0]
  end

  def test_instance_classify_empty
    results = Classifier.instance.classify("")
    assert results.first[1] < 0.5, results.first.inspect
  end

  def test_gc
    Classifier.instance.gc
  end

  # def test_instance_classify
  #   Sample.each do |sample|
  #     results = Classifier.instance.classify(sample.data)
  #     assert_equal sample.language, results.first[0], sample.path
  #   end
  # end
end