]> gitweb.fperrin.net Git - DictionaryPC.git/blobdiff - src/com/hughes/android/dictionary/engine/DictionaryTest.java
Skip Italian references.
[DictionaryPC.git] / src / com / hughes / android / dictionary / engine / DictionaryTest.java
index 53b3a916f692f303075ac0e5c16d39b4ffeea449..3de220b0db7de98609172b8aa543904142fdad54 100644 (file)
@@ -16,18 +16,21 @@ package com.hughes.android.dictionary.engine;
 
 import java.io.IOException;
 import java.io.RandomAccessFile;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Random;
 import java.util.concurrent.atomic.AtomicBoolean;
 
 import junit.framework.TestCase;
 
 import com.hughes.android.dictionary.engine.Index.IndexEntry;
-import com.hughes.util.ListUtil;
+import com.hughes.util.CollectionUtil;
 
 
 public class DictionaryTest extends TestCase {
   
   static final String TEST_OUTPUTS = com.hughes.android.dictionary.engine.DictionaryBuilderTest.TEST_OUTPUTS;
-  public static final String OUTPUTS = "../DictionaryData/outputs/";
+  public static final String OUTPUTS = "data/outputs/";
 
   @Override
   protected void setUp() {
@@ -39,32 +42,82 @@ public class DictionaryTest extends TestCase {
       }
     }
   }
+  
+  public void testURLFormatting() {
+  }
 
   public void testEnItWiktionary() throws IOException {
-    final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "EN-IT_enwiktionary.quickdic", "r");
+    final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "EN-IT.quickdic", "r");
     final Dictionary dict = new Dictionary(raf);
     final Index enIndex = dict.indices.get(0);
     
     final RowBase row = enIndex.rows.get(4);
-    assertEquals("eagle (A gold coin with a face value of $10.00) (noun)\tmoneta di dieci dollari", row.getRawText(false));
+    assertEquals("-ical", row.getRawText(false));
+    
+    final Index itIndex = dict.indices.get(1);
+    {
+    final List<RowBase> rows = itIndex.multiWordSearch("come mai", Arrays.asList("come", "mai"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertTrue(rows.get(0).toString().startsWith("come mai@"));
+    assertTrue(rows.get(0) instanceof TokenRow);
+    assertTrue(!((TokenRow)rows.get(0)).getIndexEntry().htmlEntries.isEmpty());
+    }
+
+    {
+        final IndexEntry searchResult = itIndex.findInsertionPoint("azzurro", new AtomicBoolean(
+                false));
+        HtmlEntry htmlEntry = searchResult.htmlEntries.get(0);
+        System.out.println("azzurro:\n" + htmlEntry.getHtml());
+    }
 
     raf.close();
   }
 
+  public void testFr() throws IOException {
+      final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "FR.quickdic", "r");
+      final Dictionary dict = new Dictionary(raf);
+      final Index frIndex = dict.indices.get(0);
+      
+      // Now they're all cached, we shouldn't have to search.
+//      for (final IndexEntry indexEntry : frIndex.sortedIndexEntries) {
+//          System.out.println(indexEntry.token);
+//      }
+
+      raf.close();
+  }
+
+  
+  public void testDeEnWiktionary() throws IOException {
+      final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "DE-EN.quickdic", "r");
+      final Dictionary dict = new Dictionary(raf);
+            
+      final Index deIndex = dict.indices.get(0);
+
+      {
+          final IndexEntry searchResult = deIndex.findInsertionPoint("rot", new AtomicBoolean(
+                  false));
+          HtmlEntry htmlEntry = searchResult.htmlEntries.get(0);
+          System.out.println("rot:\n" + htmlEntry.getHtml());
+      }
+
+      raf.close();
+    }
+
   public void testGermanMetadata() throws IOException {
     final RandomAccessFile raf = new RandomAccessFile(TEST_OUTPUTS + "de-en.quickdic", "r");
     final Dictionary dict = new Dictionary(raf);
     final Index deIndex = dict.indices.get(0);
     
-    assertEquals("de", deIndex.shortName);
-    assertEquals("de->en", deIndex.longName);
+    assertEquals("DE", deIndex.shortName);
+    assertEquals("DE->EN", deIndex.longName);
     
     assertEquals(2, dict.sources.size());
     assertEquals("chemnitz", dict.sources.get(0).name);
     assertEquals("dictcc", dict.sources.get(1).name);
     
-    assertEquals("chemnitz", dict.pairEntries.get(0).entrySource.name);
-    assertEquals("dictcc", ListUtil.getLast(dict.pairEntries).entrySource.name);
+    assertEquals("dictcc", dict.pairEntries.get(0).entrySource.name);
+    assertEquals("chemnitz", dict.pairEntries.get(1).entrySource.name);
     
     raf.close();
   }
@@ -151,15 +204,185 @@ public class DictionaryTest extends TestCase {
   }
   
   public void testChemnitz() throws IOException {
-    final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "/de-en_chemnitz.quickdic", "r");
+    final RandomAccessFile raf = new RandomAccessFile(TEST_OUTPUTS + "de-en.quickdic", "r");
     final Dictionary dict = new Dictionary(raf);
     final Index deIndex = dict.indices.get(0);
     
     assertSearchResult("Höschen", "Hos", deIndex.findInsertionPoint("Hos", new AtomicBoolean(false)));
     assertSearchResult("Höschen", "hos", deIndex.findInsertionPoint("hos", new AtomicBoolean(false)));
+    
+    raf.close();
+  }
+
+  public void testMultiSearch() throws IOException {
+    final RandomAccessFile raf = new RandomAccessFile(TEST_OUTPUTS + "de-en.quickdic", "r");
+    final Dictionary dict = new Dictionary(raf);
+    final Index deIndex = dict.indices.get(0);
+
+    {
+    final List<RowBase> rows = deIndex.multiWordSearch("aaa aaab", Arrays.asList("aaa", "aaab"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    }
+    
+    raf.close();
+  }
+  
+  public void testMultiSearchIt() throws IOException {
+      final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "IT.quickdic", "r");
+      final Dictionary dict = new Dictionary(raf);
+      final Index index = dict.indices.get(0);
+
+      {
+      final List<RowBase> rows = index.multiWordSearch("fare centro", 
+              Arrays.asList("fare", "centro"), new AtomicBoolean(false));
+      System.out.println(CollectionUtil.join(rows, "\n  "));
+      assertTrue(rows.toString(), rows.size() > 0);
+      assertTrue(rows.get(0).toString().startsWith("fare centro@"));
+      }
+  }
+
+  public void testMultiSearchDeBig() throws IOException {
+    final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "DE-EN.quickdic", "r");
+    final Dictionary dict = new Dictionary(raf);
+    final Index enIndex = dict.indices.get(1);
+
+    {
+    final List<RowBase> rows = enIndex.multiWordSearch("train station", Arrays.asList("train", "station"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertTrue(rows.get(0).toString().startsWith("train station@"));
+    }
+
+    {
+    final List<RowBase> rows = enIndex.multiWordSearch("a train station", Arrays.asList("a", "train", "station"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertEquals("Bahnhofsuhr {{de-noun|g=f|plural=Bahnhofsuhren}}\tstation clock (at a train station)", rows.get(0).toString());
+    }
+
+    {
+    final List<RowBase> rows = enIndex.multiWordSearch("a station", Arrays.asList("a", "station"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertEquals("Abfahrthalle {en-noun}\tDeparture room of a station.", rows.get(0).toString());
+    }
+
+    {
+    // Should print: Giving up, too many words with prefix: p
+    final List<RowBase> rows = enIndex.multiWordSearch("p eat", Arrays.asList("p", "eat"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertTrue(rows.toString().contains("verschlingen; verputzen\tto dispatch (eat)"));
+    }
 
+    {
+    // Should print: Giving up, too many words with prefix: p
+    final List<RowBase> rows = enIndex.multiWordSearch("p p", Arrays.asList("p", "p"), new AtomicBoolean(false));
+    assertTrue(rows.size() >= 1000);
+    }
+
+    {
+    // Should print: Giving up, too many words with prefix: a
+    final List<RowBase> rows = enIndex.multiWordSearch("a a", Arrays.asList("a", "a"), new AtomicBoolean(false));
+    assertTrue(rows.size() >= 1000);
+    }
+
+    {
+    // Should print: Giving up, too many words with prefix: a
+    final List<RowBase> rows = enIndex.multiWordSearch("b ba", Arrays.asList("b", "ba"), new AtomicBoolean(false));
+    assertTrue(rows.size() >= 1000);
+    }
+
+    {
+    // Should print: Giving up, too many words with prefix: a
+    final List<RowBase> rows = enIndex.multiWordSearch("b ba", Arrays.asList("b", "ba"), new AtomicBoolean(false));
+    assertTrue(rows.size() >= 1000);
+    }
+
+    raf.close();
+  }
+
+  public void testMultiSearchBigAF() throws IOException {
+    final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "EN-AF.quickdic", "r");
+    final Dictionary dict = new Dictionary(raf);
+    final Index enIndex = dict.indices.get(0);
+
+    {
+    final List<RowBase> rows = enIndex.multiWordSearch("pig eats", Arrays.asList("pig", "eats"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertEquals("pig (someone who overeats or eats rapidly) (noun)\tvark", rows.get(0).toString());
+    }
+
+    {
+    final List<RowBase> rows = enIndex.multiWordSearch("pig eat", Arrays.asList("pig", "eat"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertEquals("pig (someone who overeats or eats rapidly) (noun)\tvark", rows.get(0).toString());
+    }
+
+    {
+    final List<RowBase> rows = enIndex.multiWordSearch("pi ea", Arrays.asList("pi", "ea"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertTrue(rows.toString().contains("pig (someone who overeats or eats rapidly) (noun)\tvark"));
+    }
+
+    {
+    final List<RowBase> rows = enIndex.multiWordSearch("p eat", Arrays.asList("p", "eat"), new AtomicBoolean(false));
+    System.out.println(CollectionUtil.join(rows, "\n  "));
+    assertTrue(rows.toString(), rows.size() > 0);
+    assertTrue(rows.toString().contains("pig (someone who overeats or eats rapidly) (noun)\tvark"));
+    }
+
+    
+    raf.close();
+  }
+
+
+  public void testExactSearch() throws IOException {
+    final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "EN-cmn.quickdic", "r");
+    final Dictionary dict = new Dictionary(raf);
+    final Index cmnIndex = dict.indices.get(1);
+
+    final Random random = new Random(10);
+    
+    for (int i = 0; i < 1000; ++i) {
+      final int ii = random.nextInt(cmnIndex.sortedIndexEntries.size());
+      final IndexEntry indexEntry = cmnIndex.sortedIndexEntries.get(ii);
+      final IndexEntry found = cmnIndex.findExact(indexEntry.token);
+      assertNotNull(found);
+      assertEquals(indexEntry.token, found.token);
+      assertEquals(indexEntry, found);  // Test of caching....
+    }
+    
     raf.close();
   }
 
+  public void testThai() throws IOException {
+    final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "EN-TH.quickdic", "r");
+    final Dictionary dict = new Dictionary(raf);
+    final Index thIndex = dict.indices.get(1);
+
+    final IndexEntry entry = thIndex.findInsertionPoint("ดี", new AtomicBoolean(false));
+    assertEquals("di", entry.token);
+    
+    raf.close();
+  }
+
+  public void testNorwegian() throws IOException {
+      final RandomAccessFile raf = new RandomAccessFile(OUTPUTS + "EN-NL.quickdic", "r");
+      final Dictionary dict = new Dictionary(raf);
+      final Index nlIndex = dict.indices.get(1);
+
+      IndexEntry entry = nlIndex.findInsertionPoint("Xhosa", new AtomicBoolean(false));
+      assertEquals("Xhosa", entry.token);
+
+      entry = nlIndex.findInsertionPoint("Zyne", new AtomicBoolean(false));
+      assertEquals("Zyne", entry.token);
+
+      raf.close();
+  }
 
 }