Changeset View
Changeset View
Standalone View
Standalone View
swh/core/tests/test_collections.py
Show First 20 Lines • Show All 63 Lines • ▼ Show 20 Lines | |||||
@parametrize | @parametrize | ||||
def test_sorted_list_iter_after__key(items): | def test_sorted_list_iter_after__key(items): | ||||
list_ = SortedList(items, key=lambda item: -item) | list_ = SortedList(items, key=lambda item: -item) | ||||
for split in items: | for split in items: | ||||
expected = reversed(sorted(item for item in items if item < split)) | expected = reversed(sorted(item for item in items if item < split)) | ||||
assert list(list_.iter_after(-split)) == list(expected), f"split: {split}" | assert list(list_.iter_after(-split)) == list(expected), f"split: {split}" | ||||
@parametrize | |||||
def test_contains(items): | |||||
list_ = SortedList() | |||||
for i in range(len(items)): | |||||
for item in items[0:i]: | |||||
assert item in list_ | |||||
for item in items[i:0]: | |||||
assert item not in list_ | |||||
list_.add(items[i]) | |||||
for item in items: | |||||
assert item in list_ |