Add another doctest

This commit is contained in:
Jakob Moser 2025-05-21 10:17:50 +02:00
parent d3d28777e7
commit e71cced6e4
Signed by: jakob
GPG Key ID: 3EF2BA2851B3F53C

View File

@ -13,6 +13,8 @@ def get_equivalence_classes[T](
>>> from math import floor
>>> get_equivalence_classes([1.0, 2.1, 1.1, 1.0, 2.2, 2.1, 3.3], lambda a, b: floor(a) == floor(b))
((1.0, 1.1, 1.0), (2.1, 2.2, 2.1), (3.3,))
>>> get_equivalence_classes(range(15), lambda a, b: a % 3 == b % 3)
((0, 3, 6, 9, 12), (1, 4, 7, 10, 13), (2, 5, 8, 11, 14))
"""
equivalence_classes: list[list[T]] = []