55 lines
1.1 KiB
Python
55 lines
1.1 KiB
Python
import unittest
|
|
|
|
from src.great_ai.utilities.unique import unique
|
|
|
|
|
|
original = [
|
|
("a", 1),
|
|
("b", 5),
|
|
("a", -3),
|
|
("c", 5),
|
|
("d", 2),
|
|
("a", 2),
|
|
]
|
|
|
|
|
|
class TestUnique(unittest.TestCase):
|
|
def test_default(self) -> None:
|
|
values = [
|
|
*original,
|
|
("a", 2),
|
|
("a", 1),
|
|
("a", -3),
|
|
("b", 5),
|
|
("c", 5),
|
|
("d", 2),
|
|
]
|
|
|
|
expected = original
|
|
|
|
self.assertEqual(unique(values), expected)
|
|
self.assertEqual(unique(values, key=lambda v: v), expected)
|
|
|
|
def test_with_key_0(self) -> None:
|
|
values = original
|
|
|
|
expected = [
|
|
("a", 1),
|
|
("b", 5),
|
|
("c", 5),
|
|
("d", 2),
|
|
]
|
|
|
|
self.assertEqual(unique(values, key=lambda v: v[0]), expected)
|
|
|
|
def test_with_key_1(self) -> None:
|
|
values = original
|
|
|
|
expected = [
|
|
("a", 1),
|
|
("b", 5),
|
|
("a", -3),
|
|
("d", 2),
|
|
]
|
|
|
|
self.assertEqual(unique(values, key=lambda v: v[1]), expected)
|