| /* |
| * 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 org.apache.lucene.analysis.br; |
| |
| |
| import java.io.IOException; |
| import java.io.StringReader; |
| |
| import org.apache.lucene.analysis.Analyzer; |
| import org.apache.lucene.analysis.BaseTokenStreamTestCase; |
| import org.apache.lucene.analysis.CharArraySet; |
| import org.apache.lucene.analysis.Tokenizer; |
| import org.apache.lucene.analysis.core.KeywordTokenizer; |
| import org.apache.lucene.analysis.core.LetterTokenizer; |
| import org.apache.lucene.analysis.core.LowerCaseFilter; |
| import org.apache.lucene.analysis.miscellaneous.SetKeywordMarkerFilter; |
| |
| /** |
| * Test the Brazilian Stem Filter, which only modifies the term text. |
| * |
| * It is very similar to the snowball portuguese algorithm but not exactly the same. |
| * |
| */ |
| public class TestBrazilianAnalyzer extends BaseTokenStreamTestCase { |
| |
| public void testWithSnowballExamples() throws Exception { |
| check("boa", "boa"); |
| check("boainain", "boainain"); |
| check("boas", "boas"); |
| check("bôas", "boas"); // removes diacritic: different from snowball portugese |
| check("boassu", "boassu"); |
| check("boataria", "boat"); |
| check("boate", "boat"); |
| check("boates", "boat"); |
| check("boatos", "boat"); |
| check("bob", "bob"); |
| check("boba", "bob"); |
| check("bobagem", "bobag"); |
| check("bobagens", "bobagens"); |
| check("bobalhões", "bobalho"); // removes diacritic: different from snowball portugese |
| check("bobear", "bob"); |
| check("bobeira", "bobeir"); |
| check("bobinho", "bobinh"); |
| check("bobinhos", "bobinh"); |
| check("bobo", "bob"); |
| check("bobs", "bobs"); |
| check("boca", "boc"); |
| check("bocadas", "boc"); |
| check("bocadinho", "bocadinh"); |
| check("bocado", "boc"); |
| check("bocaiúva", "bocaiuv"); // removes diacritic: different from snowball portuguese |
| check("boçal", "bocal"); // removes diacritic: different from snowball portuguese |
| check("bocarra", "bocarr"); |
| check("bocas", "boc"); |
| check("bode", "bod"); |
| check("bodoque", "bodoqu"); |
| check("body", "body"); |
| check("boeing", "boeing"); |
| check("boem", "boem"); |
| check("boemia", "boem"); |
| check("boêmio", "boemi"); // removes diacritic: different from snowball portuguese |
| check("bogotá", "bogot"); |
| check("boi", "boi"); |
| check("bóia", "boi"); // removes diacritic: different from snowball portuguese |
| check("boiando", "boi"); |
| check("quiabo", "quiab"); |
| check("quicaram", "quic"); |
| check("quickly", "quickly"); |
| check("quieto", "quiet"); |
| check("quietos", "quiet"); |
| check("quilate", "quilat"); |
| check("quilates", "quilat"); |
| check("quilinhos", "quilinh"); |
| check("quilo", "quil"); |
| check("quilombo", "quilomb"); |
| check("quilométricas", "quilometr"); // removes diacritic: different from snowball portuguese |
| check("quilométricos", "quilometr"); // removes diacritic: different from snowball portuguese |
| check("quilômetro", "quilometr"); // removes diacritic: different from snowball portoguese |
| check("quilômetros", "quilometr"); // removes diacritic: different from snowball portoguese |
| check("quilos", "quil"); |
| check("quimica", "quimic"); |
| check("quilos", "quil"); |
| check("quimica", "quimic"); |
| check("quimicas", "quimic"); |
| check("quimico", "quimic"); |
| check("quimicos", "quimic"); |
| check("quimioterapia", "quimioterap"); |
| check("quimioterápicos", "quimioterap"); // removes diacritic: different from snowball portoguese |
| check("quimono", "quimon"); |
| check("quincas", "quinc"); |
| check("quinhão", "quinha"); // removes diacritic: different from snowball portoguese |
| check("quinhentos", "quinhent"); |
| check("quinn", "quinn"); |
| check("quino", "quin"); |
| check("quinta", "quint"); |
| check("quintal", "quintal"); |
| check("quintana", "quintan"); |
| check("quintanilha", "quintanilh"); |
| check("quintão", "quinta"); // removes diacritic: different from snowball portoguese |
| check("quintessência", "quintessente"); // versus snowball portuguese 'quintessent' |
| check("quintino", "quintin"); |
| check("quinto", "quint"); |
| check("quintos", "quint"); |
| check("quintuplicou", "quintuplic"); |
| check("quinze", "quinz"); |
| check("quinzena", "quinzen"); |
| check("quiosque", "quiosqu"); |
| } |
| |
| public void testNormalization() throws Exception { |
| check("Brasil", "brasil"); // lowercase by default |
| check("Brasília", "brasil"); // remove diacritics |
| check("quimio5terápicos", "quimio5terapicos"); // contains non-letter, diacritic will still be removed |
| check("áá", "áá"); // token is too short: diacritics are not removed |
| check("ááá", "aaa"); // normally, diacritics are removed |
| } |
| |
| public void testReusableTokenStream() throws Exception { |
| Analyzer a = new BrazilianAnalyzer(); |
| checkReuse(a, "boa", "boa"); |
| checkReuse(a, "boainain", "boainain"); |
| checkReuse(a, "boas", "boas"); |
| checkReuse(a, "bôas", "boas"); // removes diacritic: different from snowball portugese |
| a.close(); |
| } |
| |
| public void testStemExclusionTable() throws Exception { |
| BrazilianAnalyzer a = new BrazilianAnalyzer( |
| CharArraySet.EMPTY_SET, new CharArraySet(asSet("quintessência"), false)); |
| checkReuse(a, "quintessência", "quintessência"); // excluded words will be completely unchanged. |
| a.close(); |
| } |
| |
| public void testWithKeywordAttribute() throws IOException { |
| CharArraySet set = new CharArraySet(1, true); |
| set.add("Brasília"); |
| Tokenizer tokenizer = new LetterTokenizer(); |
| tokenizer.setReader(new StringReader("Brasília Brasilia")); |
| BrazilianStemFilter filter = new BrazilianStemFilter(new SetKeywordMarkerFilter(new LowerCaseFilter(tokenizer), set)); |
| |
| assertTokenStreamContents(filter, new String[] { "brasília", "brasil" }); |
| } |
| |
| private void check(final String input, final String expected) throws Exception { |
| BrazilianAnalyzer a = new BrazilianAnalyzer(); |
| checkOneTerm(a, input, expected); |
| a.close(); |
| } |
| |
| private void checkReuse(Analyzer a, String input, String expected) throws Exception { |
| checkOneTerm(a, input, expected); |
| } |
| |
| /** blast some random strings through the analyzer */ |
| public void testRandomStrings() throws Exception { |
| BrazilianAnalyzer a = new BrazilianAnalyzer(); |
| checkRandomData(random(), a, 200 * RANDOM_MULTIPLIER); |
| a.close(); |
| } |
| |
| public void testEmptyTerm() throws IOException { |
| Analyzer a = new Analyzer() { |
| @Override |
| protected TokenStreamComponents createComponents(String fieldName) { |
| Tokenizer tokenizer = new KeywordTokenizer(); |
| return new TokenStreamComponents(tokenizer, new BrazilianStemFilter(tokenizer)); |
| } |
| }; |
| checkOneTerm(a, "", ""); |
| a.close(); |
| } |
| } |