1.1 --- a/test.py Sat Sep 05 18:10:50 2009 +0200
1.2 +++ b/test.py Sun Sep 06 02:01:00 2009 +0200
1.3 @@ -201,8 +201,8 @@
1.4 doc_fields_reversed = doc_fields[:]
1.5 doc_fields_reversed.reverse()
1.6 for docnum, fields in doc_fields_reversed:
1.7 - df = rd.get_fields(docnum)
1.8 - print list(enumerate(fields)) == df, list(enumerate(fields)), df
1.9 + df = dict(rd.get_fields(docnum))
1.10 + print dict(enumerate(fields)) == df, dict(enumerate(fields)), df
1.11 for docnum in (13579, 246810):
1.12 df = rd.get_fields(docnum)
1.13 print df is None, df
1.14 @@ -309,6 +309,13 @@
1.15 for term in ("dog", "dingo"):
1.16 t = rd._find_term(term)
1.17 print t is None, t
1.18 +
1.19 +# (Test term prefix searching.)
1.20 +
1.21 +print rd.find_terms("a") == ["aardvark", "anteater"], rd.find_terms("a"), ["aardvark", "anteater"]
1.22 +print rd.find_terms("bu") == ["bull", "bulldog"], rd.find_terms("bu"), ["bull", "bulldog"]
1.23 +print rd.find_terms("c") == ["cat"], rd.find_terms("c"), ["cat"]
1.24 +print rd.find_terms("d") == [], rd.find_terms("d"), []
1.25 rd.close()
1.26
1.27 # Test dictionaries with term and position data.
1.28 @@ -361,7 +368,7 @@
1.29 for term, doc_positions in terms_reversed:
1.30 dp = list(rd.find_positions(term))
1.31 print doc_positions == dp, doc_positions, dp
1.32 -for term in ("dog", "dingo"):
1.33 +for term in ("aaa", "dog", "dingo"):
1.34 dp = rd.find_positions(term)
1.35 print dp is None, dp
1.36
1.37 @@ -422,8 +429,8 @@
1.38 fr = rd.get_frequency(term)
1.39 print frequency == fr, frequency, fr
1.40 for docnum, text in docs:
1.41 - df = rd.get_fields(docnum)
1.42 - print (123, text) == df[0], (123, text), df[0]
1.43 + df = dict(rd.get_fields(docnum))
1.44 + print df[123] == text, text, df[123]
1.45 for term, docnum, positions in position_tests:
1.46 dp = rd.find_positions(term)
1.47 pos = dp.from_document(docnum)