def test_ordered_pairs():
neighbors = [
[1, 3],
[1, 2],
[2, 5],
[3, 2],
[4, 5],
]
images = [1, 2, 3]
pairs = pairs_selection.ordered_pairs(neighbors, images)
assert set(pairs) == {(1, 2), (1, 3), (2, 5), (3, 2)}
def test_ordered_pairs() -> None:
neighbors = [
[1, 3],
[1, 2],
[2, 5],
[3, 2],
[4, 5],
]
images = [1, 2, 3]
# pyre-fixme[6]: For 1st param expected `Set[Tuple[str, str]]` but got
# `List[List[int]]`.
# pyre-fixme[6]: For 2nd param expected `List[str]` but got `List[int]`.
pairs = pairs_selection.ordered_pairs(neighbors, images)
assert set(pairs) == {(1, 2), (1, 3), (2, 5), (3, 2)}
