diff options
Diffstat (limited to 'test/test_classifier.rb')
| -rw-r--r-- | test/test_classifier.rb | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/test/test_classifier.rb b/test/test_classifier.rb index a24c5ba..0a47783 100644 --- a/test/test_classifier.rb +++ b/test/test_classifier.rb @@ -54,11 +54,8 @@ class TestClassifier < Test::Unit::TestCase def test_classify_ambiguous_languages Samples.each do |sample| - language = Linguist::Language.find_by_name(sample[:language]) - next unless language.overrides.any? - - extname = File.extname(sample[:path]) - languages = Language.all.select { |l| l.extensions.include?(extname) }.map(&:name) + 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::DATA, File.read(sample[:path]), languages) |