mirror of
https://github.com/KevinMidboe/linguist.git
synced 2025-10-29 17:50:22 +00:00
65296e86a3
This gives us a consistent test framework across all Ruby versions which should help avoid errors that are only found when CI runs the tests on different Rubies. (And this fixes an immediate bug where there's no `skip` method in the version of test-unit we're currently using only on Ruby 2.2.)
61 lines
1.9 KiB
Ruby
61 lines
1.9 KiB
Ruby
require_relative "./helper"
|
|
|
|
class TestClassifier < Minitest::Test
|
|
include Linguist
|
|
|
|
def samples_path
|
|
File.expand_path("../../samples", __FILE__)
|
|
end
|
|
|
|
def fixture(name)
|
|
File.read(File.join(samples_path, name))
|
|
end
|
|
|
|
def test_classify
|
|
db = {}
|
|
Classifier.train! db, "Ruby", fixture("Ruby/foo.rb")
|
|
Classifier.train! db, "Objective-C", fixture("Objective-C/Foo.h")
|
|
Classifier.train! db, "Objective-C", fixture("Objective-C/Foo.m")
|
|
|
|
results = Classifier.classify(db, fixture("Objective-C/hello.m"))
|
|
assert_equal "Objective-C", results.first[0]
|
|
|
|
tokens = Tokenizer.tokenize(fixture("Objective-C/hello.m"))
|
|
results = Classifier.classify(db, tokens)
|
|
assert_equal "Objective-C", results.first[0]
|
|
end
|
|
|
|
def test_restricted_classify
|
|
db = {}
|
|
Classifier.train! db, "Ruby", fixture("Ruby/foo.rb")
|
|
Classifier.train! db, "Objective-C", fixture("Objective-C/Foo.h")
|
|
Classifier.train! db, "Objective-C", fixture("Objective-C/Foo.m")
|
|
|
|
results = Classifier.classify(db, fixture("Objective-C/hello.m"), ["Objective-C"])
|
|
assert_equal "Objective-C", results.first[0]
|
|
|
|
results = Classifier.classify(db, fixture("Objective-C/hello.m"), ["Ruby"])
|
|
assert_equal "Ruby", results.first[0]
|
|
end
|
|
|
|
def test_instance_classify_empty
|
|
results = Classifier.classify(Samples.cache, "")
|
|
assert results.first[1] < 0.5, results.first.inspect
|
|
end
|
|
|
|
def test_instance_classify_nil
|
|
assert_equal [], Classifier.classify(Samples.cache, nil)
|
|
end
|
|
|
|
def test_classify_ambiguous_languages
|
|
Samples.each do |sample|
|
|
language = Linguist::Language.find_by_name(sample[:language])
|
|
languages = Language.find_by_filename(sample[:path]).map(&:name)
|
|
next unless languages.length > 1
|
|
|
|
results = Classifier.classify(Samples.cache, File.read(sample[:path]), languages)
|
|
assert_equal language.name, results.first[0], "#{sample[:path]}\n#{results.inspect}"
|
|
end
|
|
end
|
|
end
|