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

        assert unique(values) == expected
        assert 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),
        ]

        assert 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),
        ]

        assert unique(values, key=lambda v: v[1]) == expected