-
Notifications
You must be signed in to change notification settings - Fork 2
/
11720.py
42 lines (32 loc) · 1.03 KB
/
11720.py
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
# [ 백준 ] 11720번: 숫자의 합
def solution() -> None:
N: int = int(input())
print(sum(map(int, list(input()))))
if __name__ == "__main__":
from io import StringIO
from unittest.mock import patch
def test_example_case(input: list[str]) -> str:
with patch("builtins.input", side_effect=input):
with patch("sys.stdout", new_callable=StringIO) as test_stdout:
solution()
return test_stdout.getvalue()
cases: list[dict[str, list[str] | str]] = [
{
"input": ["1", "1"],
"output": "1\n"
},
{
"input": ["5", "54321"],
"output": "15\n"
},
{
"input": ["25", "7000000000000000000000000"],
"output": "7\n"
},
{
"input": ["11", "10987654321"],
"output": "46\n"
}
]
for case in cases:
assert case["output"] == test_example_case(input=case["input"])