zhaih commented on code in PR #12208:
URL: https://github.com/apache/lucene/pull/12208#discussion_r1158810796
##########
lucene/sandbox/src/test/org/apache/lucene/sandbox/search/TestTermAutomatonQuery.java:
##########
@@ -842,6 +844,95 @@ public void testRewriteSimplePhrase() throws Exception {
IOUtils.close(w, r, dir);
}
+ public void testExplainNonMatchingDocument() throws Exception {
+ TermAutomatonQuery q = new TermAutomatonQuery("field");
+ int initState = q.createState();
+ int s1 = q.createState();
+ int s2 = q.createState();
+ q.addTransition(initState, s1, "xml");
+ q.addTransition(s1, s2, "json");
+ q.setAccept(s2, true);
+ q.finish();
+
+ Directory dir = newDirectory();
+ RandomIndexWriter w = new RandomIndexWriter(random(), dir);
+ Document doc = new Document();
+ doc.add(newTextField("field", "protobuf", Field.Store.NO));
+ w.addDocument(doc);
+
+ IndexReader r = w.getReader();
+ IndexSearcher searcher = newSearcher(r);
+ Query rewrite = q.rewrite(searcher);
+ assertTrue(rewrite instanceof PhraseQuery);
+
+ TopDocs topDocs = searcher.search(rewrite, 10);
+ assertEquals(0, topDocs.totalHits.value);
+
+ for (ScoreDoc scoreDoc : topDocs.scoreDocs) {
+ Explanation explanation = searcher.explain(rewrite, scoreDoc.doc);
+ assertNotNull("Explanation should not be null", explanation);
+ assertTrue(
+ "Explanation score should match the actual score",
+ Float.compare(scoreDoc.score, (Float) explanation.getValue()) == 0);
+ }
+
+ IOUtils.close(w, r, dir);
+ }
+
+ // TODO: improve experience of working with explain
+
+ public void testExplainMatchingDocuments() throws Exception {
+ TermAutomatonQuery q = new TermAutomatonQuery("field");
+
+ int initState = q.createState();
+ int s1 = q.createState();
+ int s2 = q.createState();
+ int s3 = q.createState();
+ q.addTransition(initState, s1, "foo");
+ q.addAnyTransition(s1, s2);
+ q.addTransition(s2, s3, "bar");
+ q.setAccept(s3, true);
+ q.finish();
+
+ Directory dir = newDirectory();
+ RandomIndexWriter w = new RandomIndexWriter(random(), dir);
+ Document doc1 = new Document();
+ doc1.add(newTextField("field", "foo x bar", Field.Store.NO));
+ w.addDocument(doc1);
+
+ Document doc2 = new Document();
+ doc2.add(newTextField("field", "foo y bar", Field.Store.NO));
+ w.addDocument(doc2);
+
+ IndexReader r = w.getReader();
+ IndexSearcher searcher = newSearcher(r);
+
+ QueryUtils.check(random(), q, searcher);
+ TopDocs topDocs = searcher.search(q, 10);
+ // Verify that the docs matched fasho
+ assertEquals(2, topDocs.totalHits.value);
+
+ // Iterate through score docs and assert aspects of explanation
+ for (ScoreDoc scoreDoc : topDocs.scoreDocs) {
+ Explanation explanation = searcher.explain(q, scoreDoc.doc);
Review Comment:
Oh I just want to make sure we're testing the target class.
Since I think this test should test specifically the explain impl of
`TermAutomtonQuery` but not `PhraseQuery` or `MultiPhraseQuery` as that should
be tested somewhere else.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: issues-unsubscr...@lucene.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@lucene.apache.org
For additional commands, e-mail: issues-h...@lucene.apache.org
