andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Update test style

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-05 13:14:05 +02:00
parent f188724d4b
commit 2db2253578
18 changed files with 778 additions and 686 deletions
Download patch file Download diff file Expand all files Collapse all files

67
tests/utilities/test_unique.py
View file

@ -1,5 +1,3 @@
import unittest
from great_ai.utilities import unique
original = [
@ -12,43 +10,44 @@ original = [
]
class TestUnique(unittest.TestCase):
def test_default(self) -> None:
values = [
*original,
("a", 2),
("a", 1),
("a", -3),
("b", 5),
("c", 5),
("d", 2),
]
def test_default() -> None:
values = [
*original,
("a", 2),
("a", 1),
("a", -3),
("b", 5),
("c", 5),
("d", 2),
]
expected = original
expected = original
assert unique(values) == expected
assert unique(values, key=lambda v: v) == expected
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),
]
def test_with_key_0() -> None:
values = original
assert unique(values, key=lambda v: v[0]) == expected
expected = [
("a", 1),
("b", 5),
("c", 5),
("d", 2),
]
def test_with_key_1(self) -> None:
values = original
assert unique(values, key=lambda v: v[0]) == expected
expected = [
("a", 1),
("b", 5),
("a", -3),
("d", 2),
]
assert unique(values, key=lambda v: v[1]) == expected
def test_with_key_1() -> None:
values = original
expected = [
("a", 1),
("b", 5),
("a", -3),
("d", 2),
]
assert unique(values, key=lambda v: v[1]) == expected