1.1 --- a/test.py Thu Sep 03 01:09:06 2009 +0200
1.2 +++ b/test.py Thu Sep 03 22:10:43 2009 +0200
1.3 @@ -322,6 +322,16 @@
1.4 ("cat", [(123, [12, 145, 196]), (1200, [113])])
1.5 ]
1.6
1.7 +position_dict_tests = [
1.8 + ("badger", 19, [55, 1333]),
1.9 + ("badger", 20, None),
1.10 + ("bull", 6, [128]),
1.11 + ("bull", 26, [1, 3, 5, 7, 9]),
1.12 + ("cat", 111, None),
1.13 + ("cat", 123, [12, 145, 196]),
1.14 + ("cat", 1234, None)
1.15 + ]
1.16 +
1.17 f = open("test", "wb")
1.18 w = iixr.TermWriter(f)
1.19 f2 = open("testI", "wb")
1.20 @@ -355,6 +365,13 @@
1.21 dp = rd.find_positions(term)
1.22 print dp is None, dp
1.23
1.24 +# (Test iterators.)
1.25 +
1.26 +for term, docnum, positions in position_dict_tests:
1.27 + dp = rd.find_positions(term)
1.28 + pos = dp.from_document(docnum)
1.29 + print positions is None and pos is None or pos is not None and positions == list(pos), positions, pos
1.30 +
1.31 # (Test sequential access.)
1.32
1.33 rd.rewind()
1.34 @@ -386,7 +403,8 @@
1.35 position_tests = [
1.36 ("Every", 14, [0]),
1.37 ("sea", 36, [2, 6]),
1.38 - ("shells", 1, None)
1.39 + ("shells", 1, None),
1.40 + ("shells", 37, None)
1.41 ]
1.42
1.43 index = iixr.Index("test_index")
1.44 @@ -408,8 +426,8 @@
1.45 print (123, text) == df[0], (123, text), df[0]
1.46 for term, docnum, positions in position_tests:
1.47 dp = rd.find_positions(term)
1.48 - pos = dp[docnum]
1.49 - print positions is None and positions is pos or positions == list(pos), positions, pos
1.50 + pos = dp.from_document(docnum)
1.51 + print positions is None and pos is None or pos is not None and positions == list(pos), positions, pos
1.52 index.close()
1.53
1.54 # vim: tabstop=4 expandtab shiftwidth=4