| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 1 | import java.util.*; |
| 2 | import java.io.*; |
| 3 | |
| 4 | import org.apache.lucene.util.Version; |
| 5 | import org.apache.lucene.util.BytesRef; |
| 6 | import org.apache.lucene.util.Bits; |
| 7 | |
| 8 | import static org.junit.Assert.*; |
| 9 | import org.junit.Test; |
| 10 | import org.junit.Ignore; |
| 11 | import org.junit.runner.RunWith; |
| 12 | import org.junit.runners.JUnit4; |
| 13 | |
| 14 | import de.ids_mannheim.korap.KorapFilter; |
| 15 | |
| 16 | @RunWith(JUnit4.class) |
| 17 | public class TestKorapFilter { |
| 18 | |
| 19 | @Test |
| 20 | public void filterExample () throws IOException { |
| 21 | |
| 22 | KorapFilter kf = new KorapFilter(); |
| 23 | |
| Nils Diewald | baf68c5 | 2013-11-20 13:22:19 +0000 | [diff] [blame] | 24 | assertEquals("+textClass:tree", kf.and("textClass","tree").toString()); |
| 25 | assertEquals("+textClass:tree +textClass:sport", kf.and("textClass","tree").and("textClass","sport").toString()); |
| 26 | assertEquals("+textClass:tree +textClass:sport textClass:news", kf.and("textClass","tree").and("textClass","sport").or("textClass","news").toString()); |
| 27 | assertEquals("+textClass:tree +textClass:sport +textClass:news", kf.and("textClass", "tree", "sport", "news").toString()); |
| Nils Diewald | fb4d7b0 | 2014-04-09 17:56:17 +0000 | [diff] [blame^] | 28 | |
| 29 | assertEquals("corpusID:c-1 corpusID:c-2 corpusID:c-3", kf.or("corpusID", "c-1", "c-2", "c-3").toString()); |
| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 30 | }; |
| 31 | |
| 32 | @Test |
| 33 | public void rangeExample () throws IOException { |
| 34 | |
| 35 | KorapFilter kf = new KorapFilter(); |
| 36 | |
| Nils Diewald | baf68c5 | 2013-11-20 13:22:19 +0000 | [diff] [blame] | 37 | assertEquals("+pubDate:[20030604 TO 20030899]", kf.between("2003-06-04", "2003-08-99").toString()); |
| 38 | assertEquals("+pubDate:[0 TO 20030604]", kf.till("2003-06-04").toString()); |
| 39 | assertEquals("+pubDate:[20030604 TO 99999999]", kf.since("2003-06-04").toString()); |
| 40 | assertEquals("+pubDate:20030604", kf.date("2003-06-04").toString()); |
| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 41 | }; |
| 42 | |
| 43 | @Test |
| 44 | public void rangeLimited () throws IOException { |
| 45 | |
| 46 | KorapFilter kf = new KorapFilter(); |
| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 47 | |
| Nils Diewald | baf68c5 | 2013-11-20 13:22:19 +0000 | [diff] [blame] | 48 | assertEquals("+pubDate:[20050000 TO 20099999]", kf.between("2005", "2009").toString()); |
| 49 | assertEquals("+pubDate:[20051000 TO 20090899]", kf.between("200510", "200908").toString()); |
| 50 | assertEquals("+pubDate:[20051000 TO 20090899]", kf.between("2005-10", "2009-08").toString()); |
| 51 | assertEquals("+pubDate:[20051006 TO 20090803]", kf.between("2005-1006", "2009-0803").toString()); |
| 52 | assertEquals("+pubDate:[20051006 TO 20090803]", kf.between("2005-10-06", "2009-08-03").toString()); |
| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 53 | |
| Nils Diewald | baf68c5 | 2013-11-20 13:22:19 +0000 | [diff] [blame] | 54 | assertEquals("+pubDate:[0 TO 20059999]", kf.till("2005").toString()); |
| 55 | assertEquals("+pubDate:[0 TO 20051099]", kf.till("200510").toString()); |
| 56 | assertEquals("+pubDate:[0 TO 20051099]", kf.till("2005-10").toString()); |
| 57 | assertEquals("+pubDate:[0 TO 20051006]", kf.till("2005-1006").toString()); |
| 58 | assertEquals("+pubDate:[0 TO 20051006]", kf.till("2005-10-06").toString()); |
| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 59 | |
| Nils Diewald | baf68c5 | 2013-11-20 13:22:19 +0000 | [diff] [blame] | 60 | assertEquals("+pubDate:[20050000 TO 99999999]", kf.since("2005").toString()); |
| 61 | assertEquals("+pubDate:[20051000 TO 99999999]", kf.since("200510").toString()); |
| 62 | assertEquals("+pubDate:[20051000 TO 99999999]", kf.since("2005-10").toString()); |
| 63 | assertEquals("+pubDate:[20051006 TO 99999999]", kf.since("2005-1006").toString()); |
| 64 | assertEquals("+pubDate:[20051006 TO 99999999]", kf.since("2005-10-06").toString()); |
| 65 | |
| 66 | assertEquals("+pubDate:[20050000 TO 20059999]", kf.date("2005").toString()); |
| 67 | assertEquals("+pubDate:[20051000 TO 20051099]", kf.date("200510").toString()); |
| 68 | assertEquals("+pubDate:[20051000 TO 20051099]", kf.date("2005-10").toString()); |
| 69 | assertEquals("+pubDate:20051006", kf.date("2005-1006").toString()); |
| 70 | assertEquals("+pubDate:20051006", kf.date("2005-10-06").toString()); |
| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 71 | }; |
| 72 | |
| 73 | @Test |
| 74 | public void rangeFailure () throws IOException { |
| 75 | |
| 76 | KorapFilter kf = new KorapFilter(); |
| Nils Diewald | baf68c5 | 2013-11-20 13:22:19 +0000 | [diff] [blame] | 77 | assertEquals("", kf.between("aaaa-bb-cc", "aaaabbcc").toString()); |
| 78 | assertEquals("", kf.till("aaaa-bb-cc").toString()); |
| 79 | assertEquals("", kf.since("aaaa-bb-cc").toString()); |
| 80 | assertEquals("", kf.date("aaaa-bb-cc").toString()); |
| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 81 | }; |
| Nils Diewald | baf68c5 | 2013-11-20 13:22:19 +0000 | [diff] [blame] | 82 | |
| 83 | |
| 84 | // TODO: More extensive testing! |
| 85 | |
| Nils Diewald | f399a67 | 2013-11-18 17:55:22 +0000 | [diff] [blame] | 86 | }; |