| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package opennlp.tools.langdetect; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| import opennlp.tools.formats.ResourceAsStreamFactory; |
| import opennlp.tools.util.PlainTextByLineStream; |
| import opennlp.tools.util.TrainingParameters; |
| |
| |
| public class LanguageDetectorMETest { |
| |
| private LanguageDetectorModel model; |
| |
| @Before |
| public void init() throws Exception { |
| |
| this.model = trainModel(); |
| |
| } |
| |
| @Test |
| public void testPredictLanguages() { |
| LanguageDetector ld = new LanguageDetectorME(this.model); |
| Language[] languages = ld.predictLanguages("estava em uma marcenaria na Rua Bruno"); |
| |
| Assert.assertEquals(4, languages.length); |
| Assert.assertEquals("pob", languages[0].getLang()); |
| Assert.assertEquals("ita", languages[1].getLang()); |
| Assert.assertEquals("spa", languages[2].getLang()); |
| Assert.assertEquals("fra", languages[3].getLang()); |
| } |
| |
| @Test |
| public void testProbingPredictLanguages() { |
| LanguageDetectorME ld = new LanguageDetectorME(this.model); |
| for (int i = 0; i < 10000; i += 1000) { |
| StringBuilder sb = new StringBuilder(); |
| for (int j = 0; j <= i; j++) { |
| sb.append("estava em uma marcenaria na Rua Bruno "); |
| } |
| ProbingLanguageDetectionResult result = ld.probingPredictLanguages(sb.toString()); |
| Assert.assertTrue(result.getLength() <= 600); |
| Language[] languages = result.getLanguages(); |
| Assert.assertEquals(4, languages.length); |
| Assert.assertEquals("pob", languages[0].getLang()); |
| Assert.assertEquals("ita", languages[1].getLang()); |
| Assert.assertEquals("spa", languages[2].getLang()); |
| Assert.assertEquals("fra", languages[3].getLang()); |
| } |
| } |
| |
| @Test |
| public void testPredictLanguage() { |
| LanguageDetector ld = new LanguageDetectorME(this.model); |
| Language language = ld.predictLanguage("Dove รจ meglio che giochi"); |
| |
| Assert.assertEquals("ita", language.getLang()); |
| } |
| |
| @Test |
| public void testSupportedLanguages() { |
| |
| LanguageDetector ld = new LanguageDetectorME(this.model); |
| String[] supportedLanguages = ld.getSupportedLanguages(); |
| |
| Assert.assertEquals(4, supportedLanguages.length); |
| } |
| |
| @Test |
| public void testLoadFromSerialized() throws IOException { |
| byte[] serialized = serializeModel(model); |
| |
| LanguageDetectorModel myModel = new LanguageDetectorModel(new ByteArrayInputStream(serialized)); |
| |
| Assert.assertNotNull(myModel); |
| |
| } |
| |
| protected static byte[] serializeModel(LanguageDetectorModel model) throws IOException { |
| |
| ByteArrayOutputStream out = new ByteArrayOutputStream(); |
| model.serialize(out); |
| return out.toByteArray(); |
| } |
| |
| public static LanguageDetectorModel trainModel() throws Exception { |
| return trainModel(new LanguageDetectorFactory()); |
| } |
| |
| public static LanguageDetectorModel trainModel(LanguageDetectorFactory factory) throws Exception { |
| |
| |
| LanguageDetectorSampleStream sampleStream = createSampleStream(); |
| |
| TrainingParameters params = new TrainingParameters(); |
| params.put(TrainingParameters.ITERATIONS_PARAM, 100); |
| params.put(TrainingParameters.CUTOFF_PARAM, 5); |
| params.put("DataIndexer", "TwoPass"); |
| params.put(TrainingParameters.ALGORITHM_PARAM, "NAIVEBAYES"); |
| |
| return LanguageDetectorME.train(sampleStream, params, factory); |
| } |
| |
| public static LanguageDetectorSampleStream createSampleStream() throws IOException { |
| |
| ResourceAsStreamFactory streamFactory = new ResourceAsStreamFactory( |
| LanguageDetectorMETest.class, "/opennlp/tools/doccat/DoccatSample.txt"); |
| |
| PlainTextByLineStream lineStream = new PlainTextByLineStream(streamFactory, "UTF-8"); |
| |
| return new LanguageDetectorSampleStream(lineStream); |
| } |
| } |