Move files
This commit is contained in:
parent
79ddc5c7df
commit
8b004ec46d
233 changed files with 117 additions and 2394 deletions
54
tests/utilities/test_unique.py
Normal file
54
tests/utilities/test_unique.py
Normal file
|
|
@ -0,0 +1,54 @@
|
|||
import unittest
|
||||
|
||||
from src.great_ai.utilities 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
|
||||
Loading…
Add table
Add a link
Reference in a new issue