1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
import pytest
from typing import NamedTuple, Optional
from rich._ratio import ratio_reduce, ratio_resolve
class Edge(NamedTuple):
size: Optional[int] = None
ratio: int = 1
minimum_size: int = 1
@pytest.mark.parametrize(
"total,ratios,maximums,values,result",
[
(20, [2, 4], [20, 20], [5, 5], [-2, -8]),
(20, [2, 4], [1, 1], [5, 5], [4, 4]),
(20, [2, 4], [1, 1], [2, 2], [1, 1]),
(3, [2, 4], [3, 3], [2, 2], [1, 0]),
(3, [2, 4], [3, 3], [0, 0], [-1, -2]),
(3, [0, 0], [3, 3], [4, 4], [4, 4]),
],
)
def test_ratio_reduce(total, ratios, maximums, values, result):
assert ratio_reduce(total, ratios, maximums, values) == result
def test_ratio_resolve():
assert ratio_resolve(100, []) == []
assert ratio_resolve(100, [Edge(size=100), Edge(ratio=1)]) == [100, 1]
assert ratio_resolve(100, [Edge(ratio=1)]) == [100]
assert ratio_resolve(100, [Edge(ratio=1), Edge(ratio=1)]) == [50, 50]
assert ratio_resolve(100, [Edge(size=20), Edge(ratio=1), Edge(ratio=1)]) == [
20,
40,
40,
]
assert ratio_resolve(100, [Edge(size=40), Edge(ratio=2), Edge(ratio=1)]) == [
40,
40,
20,
]
assert ratio_resolve(
100, [Edge(size=40), Edge(ratio=2), Edge(ratio=1, minimum_size=25)]
) == [40, 35, 25]
assert ratio_resolve(100, [Edge(ratio=1), Edge(ratio=1), Edge(ratio=1)]) == [
33,
33,
34,
]
|
