Skip to content

Commit

Permalink
move test helper into the specific class it's used in
Browse files Browse the repository at this point in the history
  • Loading branch information
hannosch committed Apr 7, 2012
1 parent 2f5aa0b commit 85ff63d
Showing 1 changed file with 14 additions and 15 deletions.
29 changes: 14 additions & 15 deletions src/Products/ZCatalog/tests/test_catalog.py
Expand Up @@ -26,15 +26,6 @@
from Products.ZCTextIndex.ZCTextIndex import ZCTextIndex


def sort(iterable, reverse=False):
L = list(iterable)
if reverse:
L.sort(reverse=True)
else:
L.sort()
return L


class zdummy(ExtensionClass.Base):
def __init__(self, num):
self.num = num
Expand Down Expand Up @@ -783,14 +774,22 @@ def _make_many(self):
catalogs.append(cat)
return catalogs, mergeResults

def _sort(self, iterable, reverse=False):
L = list(iterable)
if reverse:
L.sort(reverse=True)
else:
L.sort()
return L

def test_no_filter_or_sort(self):
catalogs, mergeResults = self._make_many()
results = [cat.searchResults(
dict(number=True), _merge=0) for cat in catalogs]
merged_rids = [r.getRID() for r in mergeResults(
results, has_sort_keys=False, reverse=False)]
expected = [r.getRID() for r in chain(*results)]
self.assertEqual(sort(merged_rids), sort(expected))
self.assertEqual(self._sort(merged_rids), self._sort(expected))

def test_sorted_only(self):
catalogs, mergeResults = self._make_many()
Expand All @@ -799,7 +798,7 @@ def test_sorted_only(self):
for cat in catalogs]
merged_rids = [r.getRID() for r in mergeResults(
results, has_sort_keys=True, reverse=False)]
expected = sort(chain(*results))
expected = self._sort(chain(*results))
expected = [rid for sortkey, rid, getitem in expected]
self.assertEqual(merged_rids, expected)

Expand All @@ -810,7 +809,7 @@ def test_sort_reverse(self):
for cat in catalogs]
merged_rids = [r.getRID() for r in mergeResults(
results, has_sort_keys=True, reverse=True)]
expected = sort(chain(*results), reverse=True)
expected = self._sort(chain(*results), reverse=True)
expected = [rid for sortkey, rid, getitem in expected]
self.assertEqual(merged_rids, expected)

Expand All @@ -822,7 +821,7 @@ def test_limit_sort(self):
for cat in catalogs]
merged_rids = [r.getRID() for r in mergeResults(
results, has_sort_keys=True, reverse=False)]
expected = sort(chain(*results))
expected = self._sort(chain(*results))
expected = [rid for sortkey, rid, getitem in expected]
self.assertEqual(merged_rids, expected)

Expand All @@ -832,7 +831,7 @@ def test_scored(self):
for cat in catalogs]
merged_rids = [r.getRID() for r in mergeResults(
results, has_sort_keys=True, reverse=False)]
expected = sort(chain(*results))
expected = self._sort(chain(*results))
expected = [rid for sortkey, (nscore, score, rid), getitem in expected]
self.assertEqual(merged_rids, expected)

Expand All @@ -844,7 +843,7 @@ def test_small_index_sort(self):
for cat in catalogs]
merged_rids = [r.getRID() for r in mergeResults(
results, has_sort_keys=True, reverse=False)]
expected = sort(chain(*results))
expected = self._sort(chain(*results))
expected = [rid for sortkey, rid, getitem in expected]
self.assertEqual(merged_rids, expected)

Expand Down

0 comments on commit 85ff63d

Please sign in to comment.