--- /dev/null
+package org.apache.lucene.analysis.br;
+
+/**
+ * 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.
+ */
+
+import java.io.IOException;
+import java.io.StringReader;
+
+import org.apache.lucene.analysis.BaseTokenStreamTestCase;
+import org.apache.lucene.analysis.Analyzer;
+import org.apache.lucene.analysis.CharArraySet;
+import org.apache.lucene.analysis.KeywordMarkerFilter;
+import org.apache.lucene.analysis.LowerCaseTokenizer;
+
+/**
+ * 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 TestBrazilianStemmer 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(TEST_VERSION_CURRENT);
+ checkReuse(a, "boa", "boa");
+ checkReuse(a, "boainain", "boainain");
+ checkReuse(a, "boas", "boas");
+ checkReuse(a, "bôas", "boas"); // removes diacritic: different from snowball portugese
+ }
+
+ public void testStemExclusionTable() throws Exception {
+ BrazilianAnalyzer a = new BrazilianAnalyzer(TEST_VERSION_CURRENT);
+ a.setStemExclusionTable(new String[] { "quintessência" });
+ checkReuse(a, "quintessência", "quintessência"); // excluded words will be completely unchanged.
+ }
+
+ public void testStemExclusionTableBWCompat() throws IOException {
+ CharArraySet set = new CharArraySet(TEST_VERSION_CURRENT, 1, true);
+ set.add("Brasília");
+ BrazilianStemFilter filter = new BrazilianStemFilter(
+ new LowerCaseTokenizer(TEST_VERSION_CURRENT, new StringReader("Brasília Brasilia")), set);
+ assertTokenStreamContents(filter, new String[] { "brasília", "brasil" });
+ }
+
+ public void testWithKeywordAttribute() throws IOException {
+ CharArraySet set = new CharArraySet(TEST_VERSION_CURRENT, 1, true);
+ set.add("Brasília");
+ BrazilianStemFilter filter = new BrazilianStemFilter(
+ new KeywordMarkerFilter(new LowerCaseTokenizer(TEST_VERSION_CURRENT, new StringReader(
+ "Brasília Brasilia")), set));
+ assertTokenStreamContents(filter, new String[] { "brasília", "brasil" });
+ }
+
+ public void testWithKeywordAttributeAndExclusionTable() throws IOException {
+ CharArraySet set = new CharArraySet(TEST_VERSION_CURRENT, 1, true);
+ set.add("Brasília");
+ CharArraySet set1 = new CharArraySet(TEST_VERSION_CURRENT, 1, true);
+ set1.add("Brasilia");
+ BrazilianStemFilter filter = new BrazilianStemFilter(
+ new KeywordMarkerFilter(new LowerCaseTokenizer(TEST_VERSION_CURRENT, new StringReader(
+ "Brasília Brasilia")), set), set1);
+ assertTokenStreamContents(filter, new String[] { "brasília", "brasilia" });
+ }
+
+ /*
+ * Test that changes to the exclusion table are applied immediately
+ * when using reusable token streams.
+ */
+ public void testExclusionTableReuse() throws Exception {
+ BrazilianAnalyzer a = new BrazilianAnalyzer(TEST_VERSION_CURRENT);
+ checkReuse(a, "quintessência", "quintessente");
+ a.setStemExclusionTable(new String[] { "quintessência" });
+ checkReuse(a, "quintessência", "quintessência");
+ }
+
+ private void check(final String input, final String expected) throws Exception {
+ checkOneTerm(new BrazilianAnalyzer(TEST_VERSION_CURRENT), input, expected);
+ }
+
+ private void checkReuse(Analyzer a, String input, String expected) throws Exception {
+ checkOneTermReuse(a, input, expected);
+ }
+
+ /** blast some random strings through the analyzer */
+ public void testRandomStrings() throws Exception {
+ checkRandomData(random, new BrazilianAnalyzer(TEST_VERSION_CURRENT), 10000*RANDOM_MULTIPLIER);
+ }
+}
\ No newline at end of file
fnp / pylucene.git / blobdiff