1.1 --- a/test.py Wed Aug 26 23:36:02 2009 +0200
1.2 +++ b/test.py Thu Aug 27 00:02:50 2009 +0200
1.3 @@ -132,10 +132,10 @@
1.4 terms_reversed = terms[:]
1.5 terms_reversed.reverse()
1.6 for term, offset in terms_reversed:
1.7 - o = rd.find(term)
1.8 + o = rd.find_term(term)
1.9 print offset == o, offset, o
1.10 for term in ("dog", "dingo"):
1.11 - o = rd.find(term)
1.12 + o = rd.find_term(term)
1.13 print o is None, o
1.14 rd.close()
1.15
1.16 @@ -176,4 +176,45 @@
1.17 print dp is None, dp
1.18 rd.close()
1.19
1.20 +docs = [
1.21 + (1, "The cat sat on the mat"),
1.22 + (2, "Every good boy deserves football"),
1.23 + (13, "One good turn deserves another"),
1.24 + (14, "Every man for himself"),
1.25 + (25, "Red sky at night shepherd's delight"),
1.26 + (36, "She sells sea shells on the sea shore")
1.27 + ]
1.28 +
1.29 +doc_tests = [
1.30 + ("Every", [(2, [0]), (14, [0])]),
1.31 + ("good", [(2, [1]), (13, [1])]),
1.32 + ("deserves", [(2, [3]), (13, [3])]),
1.33 + ("sea", [(36, [2, 6])])
1.34 + ]
1.35 +
1.36 +f = open("test", "wb")
1.37 +w = iixr.TermWriter(f)
1.38 +f2 = open("testI", "wb")
1.39 +w2 = iixr.TermIndexWriter(f2)
1.40 +f3 = open("testP", "wb")
1.41 +w3 = iixr.PositionWriter(f3)
1.42 +wd = iixr.TermDictionaryWriter(w, w2, w3, 3)
1.43 +wi = iixr.IndexWriter(wd)
1.44 +for docnum, text in docs:
1.45 + for position, term in enumerate(text.split()):
1.46 + wi.add_position(term, docnum, position)
1.47 +wi.close()
1.48 +
1.49 +f = open("test", "rb")
1.50 +r = iixr.TermReader(f)
1.51 +f2 = open("testI", "rb")
1.52 +r2 = iixr.TermIndexReader(f2)
1.53 +f3 = open("testP", "rb")
1.54 +r3 = iixr.PositionReader(f3)
1.55 +rd = iixr.TermDictionaryReader(r, r2, r3)
1.56 +for term, doc_positions in doc_tests:
1.57 + dp = rd.find_positions(term)
1.58 + print doc_positions == dp, doc_positions, dp
1.59 +rd.close()
1.60 +
1.61 # vim: tabstop=4 expandtab shiftwidth=4