Add files

Signed-off-by: András Schmelczer <andras@schmelczer.dev>

good_ai/tests/sus/test_unique.py

@@ -0,0 +1,54 @@
+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)