1.1 --- a/test.py Tue Aug 25 22:44:15 2009 +0200
1.2 +++ b/test.py Tue Aug 25 23:53:20 2009 +0200
1.3 @@ -10,7 +10,7 @@
1.4 w.write_number(number)
1.5 w.close()
1.6
1.7 -f = open("test", "r")
1.8 +f = open("test", "rb")
1.9 r = iixr.FileReader(f)
1.10 for number in numbers:
1.11 n = r.read_number()
1.12 @@ -36,7 +36,7 @@
1.13 w.reset()
1.14 w.close()
1.15
1.16 -f = open("test", "r")
1.17 +f = open("test", "rb")
1.18 r = iixr.PositionReader(f)
1.19 for doc_positions in all_doc_positions:
1.20 for docnum, positions in doc_positions:
1.21 @@ -55,7 +55,7 @@
1.22 )
1.23 w.close()
1.24
1.25 -f = open("test", "r")
1.26 +f = open("test", "rb")
1.27 r = iixr.PositionReader(f)
1.28 offsets.reverse()
1.29 all_doc_positions.reverse()
1.30 @@ -79,7 +79,7 @@
1.31 w.write_term(term, offset)
1.32 w.close()
1.33
1.34 -f = open("test", "r")
1.35 +f = open("test", "rb")
1.36 r = iixr.TermReader(f)
1.37 for term, offset in terms:
1.38 t, o = r.read_term()
1.39 @@ -87,4 +87,52 @@
1.40 print offset == o, offset, o
1.41 r.close()
1.42
1.43 +indexed_terms = [
1.44 + ("aardvark", 100000123, 200000321),
1.45 + ("anteater", 100000456, 200000654),
1.46 + ("badger", 100000789, 200000987),
1.47 + ("bull", 1000001234, 200004321),
1.48 + ("bulldog", 1000002345, 200005432),
1.49 + ("cat", 1000003456, 200006543)
1.50 + ]
1.51 +
1.52 +f = open("test", "wb")
1.53 +w = iixr.TermIndexWriter(f)
1.54 +for term, offset, info_offset in indexed_terms:
1.55 + w.write_term(term, offset, info_offset)
1.56 +w.close()
1.57 +
1.58 +f = open("test", "rb")
1.59 +r = iixr.TermIndexReader(f)
1.60 +for term, offset, info_offset in indexed_terms:
1.61 + t, o, i = r.read_term()
1.62 + print term == t, term, t
1.63 + print offset == o, offset, o
1.64 + print info_offset == i, info_offset, i
1.65 +r.close()
1.66 +
1.67 +f = open("test", "wb")
1.68 +w = iixr.TermWriter(f)
1.69 +f2 = open("testI", "wb")
1.70 +w2 = iixr.TermIndexWriter(f2)
1.71 +w3 = iixr.TermDictionaryWriter(w, w2, 3)
1.72 +for term, offset in terms:
1.73 + w3.write_term(term, offset)
1.74 +w3.close()
1.75 +
1.76 +f = open("test", "rb")
1.77 +r = iixr.TermReader(f)
1.78 +f2 = open("testI", "rb")
1.79 +r2 = iixr.TermIndexReader(f2)
1.80 +r3 = iixr.TermDictionaryReader(r, r2)
1.81 +terms_reversed = terms[:]
1.82 +terms_reversed.reverse()
1.83 +for term, offset in terms_reversed:
1.84 + o = r3.find(term)
1.85 + print offset == o, offset, o
1.86 +for term in ("dog", "dingo"):
1.87 + o = r3.find(term)
1.88 + print o is None, o
1.89 +r3.close()
1.90 +
1.91 # vim: tabstop=4 expandtab shiftwidth=4