old python needs __main__ to call a module
[pylucene.git] / samples / LuceneInAction / lia / analysis / queryparser / AnalysisParalysisTest.py
1 # ====================================================================
2 #   Licensed under the Apache License, Version 2.0 (the "License");
3 #   you may not use this file except in compliance with the License.
4 #   You may obtain a copy of the License at
5 #
6 #       http://www.apache.org/licenses/LICENSE-2.0
7 #
8 #   Unless required by applicable law or agreed to in writing, software
9 #   distributed under the License is distributed on an "AS IS" BASIS,
10 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 #   See the License for the specific language governing permissions and
12 #   limitations under the License.
13 # ====================================================================
14
15 from lia.common.LiaTestCase import LiaTestCase
16
17 from lucene import \
18     QueryParser, StandardAnalyzer, PerFieldAnalyzerWrapper, \
19     WhitespaceAnalyzer, Version
20
21
22 class AnalysisParalysisTest(LiaTestCase):
23
24     def testAnalyzer(self):
25
26         analyzer = StandardAnalyzer(Version.LUCENE_CURRENT)
27         queryString = "category:/philosophy/eastern"
28
29         parser = QueryParser(Version.LUCENE_CURRENT, "contents", analyzer)
30         parser.setAutoGeneratePhraseQueries(True)
31         query = parser.parse(queryString)
32
33         self.assertEqual("category:\"philosophy eastern\"",
34                          query.toString("contents"), "path got split, yikes!")
35
36         perFieldAnalyzer = PerFieldAnalyzerWrapper(analyzer)
37         perFieldAnalyzer.addAnalyzer("category", WhitespaceAnalyzer())
38         query = QueryParser(Version.LUCENE_CURRENT,
39                             "contents", perFieldAnalyzer).parse(queryString)
40
41         self.assertEqual("category:/philosophy/eastern",
42                          query.toString("contents"),
43                          "leave category field alone")