3 # This file is part of DisOrder.
4 # Copyright (C) 2007, 2008 Richard Kettlewell
6 # This program is free software: you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation, either version 3 of the License, or
9 # (at your option) any later version.
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 # GNU General Public License for more details.
16 # You should have received a copy of the GNU General Public License
17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
19 import dtest
,time
,disorder
,sys
23 def check_search_results(terms
, expected
):
25 # We want a consistent encoding and ordering
26 print "terms: %s" % terms
27 got
= client
.search(terms
)
28 got
= map(dtest
.nfc
, got
)
29 expected
= map(lambda s
: "%s/%s" %
(dtest
.tracks
, s
), expected
)
30 expected
= map(dtest
.nfc
, expected
)
34 print "expected: %s" % expected
40 """Check that the search produces the right results"""
45 client
= disorder
.client()
46 first
= ["Joe Bloggs/First Album/01:F\xC3\x8Crst track.ogg",
47 "Joe Bloggs/First Album/02:Second track.ogg",
48 "Joe Bloggs/First Album/03:ThI\xCC\x81rd track.ogg",
49 "Joe Bloggs/First Album/04:Fourth track.ogg",
50 "Joe Bloggs/First Album/05:Fifth track.ogg",
51 "Joe Bloggs/Second Album/01:First track.ogg",
52 "Joe Bloggs/Third Album/01:First_track.ogg"]
53 second
= ["Joe Bloggs/First Album/02:Second track.ogg",
54 "Joe Bloggs/Second Album/01:First track.ogg",
55 "Joe Bloggs/Second Album/02:Second track.ogg",
56 "Joe Bloggs/Second Album/03:Third track.ogg",
57 "Joe Bloggs/Second Album/04:Fourth track.ogg",
58 "Joe Bloggs/Second Album/05:Fifth track.ogg",
59 "Joe Bloggs/Third Album/02:Second_track.ogg"]
60 third
= ["Joe Bloggs/First Album/03:ThI\xCC\x81rd track.ogg",
61 "Joe Bloggs/Second Album/03:Third track.ogg",
62 "Joe Bloggs/Third Album/01:First_track.ogg",
63 "Joe Bloggs/Third Album/02:Second_track.ogg",
64 "Joe Bloggs/Third Album/03:Third_track.ogg",
65 "Joe Bloggs/Third Album/04:Fourth_track.ogg",
66 "Joe Bloggs/Third Album/05:Fifth_track.ogg"]
67 first_and_second
= filter(lambda s
: s
in second
, first
)
69 check_search_results(["first"], first
)
70 check_search_results(["Second"], second
)
71 check_search_results(["THIRD"], third
)
73 check_search_results(["FIRST", "SECOND"], first_and_second
)
74 # Non-ASCII Characters
75 # 00CC is LATIN CAPITAL LETTER I WITH GRAVE
76 # 00EC is LATIN SMALL LETTER I WITH GRAVE
77 check_search_results([u
"F\u00CCRST"], first
)
78 check_search_results([u
"f\u00ECrst"], first
)
79 # 00CD is LATIN CAPITAL LETTER I WITH ACUTE
80 # 00ED is LATIN SMALL LETTER I WITH ACUTE
81 check_search_results([u
"TH\u00CDRD"], third
)
82 check_search_results([u
"th\u00EDrd"], third
)
83 # ...and again in denormalized form
84 # 0300 is COMBINING GRAVE ACCENT
85 # 0301 is COMBINING ACUTE ACCENT
86 check_search_results([u
"FI\u0300RST"], first
)
87 check_search_results([u
"fi\u0300rst"], first
)
88 check_search_results([u
"THI\u0301RD"], third
)
89 check_search_results([u
"thI\u0301rd"], third
)
90 # stopwords shouldn't show up
91 check_search_results(["01"], [])
96 if __name__
== '__main__':