1.1 --- a/test.py Mon Aug 31 20:24:41 2009 +0200
1.2 +++ b/test.py Mon Aug 31 21:02:30 2009 +0200
1.3 @@ -1,6 +1,20 @@
1.4 #!/usr/bin/env python
1.5
1.6 import iixr
1.7 +import os
1.8 +
1.9 +# Remove old test files.
1.10 +
1.11 +for filename in ("test", "testF", "testFI", "testI", "testP"):
1.12 + try:
1.13 + os.remove(filename)
1.14 + except OSError:
1.15 + pass
1.16 +
1.17 +try:
1.18 + os.removedirs("test_index")
1.19 +except OSError:
1.20 + pass
1.21
1.22 # Test basic data types.
1.23
1.24 @@ -63,7 +77,7 @@
1.25 offsets.reverse()
1.26 all_doc_positions.reverse()
1.27 for offset, doc_positions in zip(offsets, all_doc_positions):
1.28 - dp = r.read_term_positions(offset)
1.29 + dp = list(r.read_term_positions(offset))
1.30 print doc_positions == dp, doc_positions, dp
1.31 r.close()
1.32
1.33 @@ -267,7 +281,7 @@
1.34 terms_reversed = terms_with_positions[:]
1.35 terms_reversed.reverse()
1.36 for term, doc_positions in terms_reversed:
1.37 - dp = rd.find_positions(term)
1.38 + dp = list(rd.find_positions(term))
1.39 print doc_positions == dp, doc_positions, dp
1.40 for term in ("dog", "dingo"):
1.41 dp = rd.find_positions(term)
1.42 @@ -278,6 +292,7 @@
1.43 rd.rewind()
1.44 for term, doc_positions in terms_with_positions:
1.45 t, fr, dp = rd.read_term()
1.46 + dp = list(dp)
1.47 print term == t, term, t
1.48 print doc_positions == dp, doc_positions, dp
1.49 rd.close()
1.50 @@ -310,7 +325,7 @@
1.51
1.52 rd = index.get_reader()
1.53 for term, frequency, doc_positions in doc_tests:
1.54 - dp = rd.find_positions(term)
1.55 + dp = list(rd.find_positions(term))
1.56 print doc_positions == dp, doc_positions, dp
1.57 fr = rd.get_frequency(term)
1.58 print frequency == fr, frequency, fr