Fix misalignment in SearchCollection and SimpleImpactSearcher impleme…

…ntations (#2165) - ref #2167

src/main/java/io/anserini/search/SimpleImpactSearcher.java

@@ -638,9 +638,8 @@ public Result[] search(Map<String, Integer> encoded_q, int k) throws IOException
   public Result[] search(String q, int k) throws IOException, OrtException {
     // make encoded query from raw query
     Map<String, Integer> encoded_q = encodeWithOnnx(q);
-    Query query = generator.buildQuery(Constants.CONTENTS, analyzer, q);
-
     String encodedQuery = encodeWithOnnx(encoded_q);
+    Query query = generator.buildQuery(Constants.CONTENTS, analyzer, encodedQuery);
     return _search(query, encodedQuery, k);
   }
 

src/test/java/io/anserini/search/SimpleImpactSearcherTest.java

@@ -370,4 +370,23 @@ public void testSearch5() throws Exception {
     isearcher_rm3.close();
   }
 
+  @Test
+  public void testSearch6() throws Exception {
+    // test the expanded query result should be diff with testSearch5
+    SimpleImpactSearcher searcher = new SimpleImpactSearcher(super.tempDir1.toString());
+    searcher.set_onnx_query_encoder("UniCoil");
+
+    Result[] results;
+
+    String query = "this is a a a a a test";
+    results = searcher.search(query, 1);
+
+    assertEquals(1, results.length);
+    assertEquals("doc3", results[0].docid);
+    assertEquals(2, results[0].lucene_docid);
+    assertEquals(130.0, results[0].score, 10e-5);
+
+    searcher.close();
+  }
+
 }