import unittest

from src.sus.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)