Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commit69aebbc

Browse files
committed
mymax: final step, 6 overloads
1 parentf5891db commit69aebbc

File tree

3 files changed

+277
-0
lines changed

3 files changed

+277
-0
lines changed

‎less_than/max_steps/99-final/mymax.py

Lines changed: 54 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
1+
fromtypingimportProtocol,Any,TypeVar,overload,Callable,Iterable,Union
2+
3+
classSupportsLessThan(Protocol):
4+
def__lt__(self,other:Any)->bool: ...
5+
6+
T=TypeVar('T')
7+
LT=TypeVar('LT',bound=SupportsLessThan)
8+
DT=TypeVar('DT')
9+
10+
MISSING=object()
11+
EMPTY_MSG='max() arg is an empty sequence'
12+
13+
@overload
14+
defmax(arg1:LT,arg2:LT,*rest:LT,key:None= ...)->LT:
15+
...
16+
@overload
17+
defmax(arg1:T,arg2:T,*rest:T,key:Callable[[T],LT])->T:
18+
...
19+
@overload
20+
defmax(iterable:Iterable[LT],*,key:None= ...)->LT:
21+
...
22+
@overload
23+
defmax(iterable:Iterable[T],*,key:Callable[[T],LT])->T:
24+
...
25+
@overload
26+
defmax(iterable:Iterable[LT],*,key:None= ...,default:DT)->Union[LT,DT]:
27+
...
28+
@overload
29+
defmax(iterable:Iterable[T],*,key:Callable[[T],LT],default:DT)->Union[T,DT]:
30+
...
31+
defmax(first,*rest,key=None,default=MISSING):
32+
ifrest:
33+
series=rest
34+
candidate=first
35+
else:
36+
series=iter(first)
37+
try:
38+
candidate=next(series)
39+
exceptStopIteration:
40+
ifdefaultisnotMISSING:
41+
returndefault
42+
raiseValueError(EMPTY_MSG)fromNone
43+
ifkeyisNone:
44+
forcurrentinseries:
45+
ifcandidate<current:
46+
candidate=current
47+
else:
48+
candidate_key=key(candidate)
49+
forcurrentinseries:
50+
current_key=key(current)
51+
ifcandidate_key<current_key:
52+
candidate=current
53+
candidate_key=current_key
54+
returncandidate
Lines changed: 127 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,127 @@
1+
fromtypingimportTYPE_CHECKING,List,Optional
2+
3+
importmymaxasmy
4+
5+
defdemo_args_list_float()->None:
6+
args= [2.5,3.5,1.5]
7+
expected=3.5
8+
result=my.max(*args)
9+
print(args,expected,result,sep='\n')
10+
assertresult==expected
11+
ifTYPE_CHECKING:
12+
reveal_type(args)
13+
reveal_type(expected)
14+
reveal_type(result)
15+
16+
defdemo_args_iter_int()->None:
17+
args= [30,10,20]
18+
expected=30
19+
result=my.max(args)
20+
print(args,expected,result,sep='\n')
21+
assertresult==expected
22+
ifTYPE_CHECKING:
23+
reveal_type(args)
24+
reveal_type(expected)
25+
reveal_type(result)
26+
27+
28+
defdemo_args_iter_str()->None:
29+
args=iter('banana kiwi mango apple'.split())
30+
expected='mango'
31+
result=my.max(args)
32+
print(args,expected,result,sep='\n')
33+
assertresult==expected
34+
ifTYPE_CHECKING:
35+
reveal_type(args)
36+
reveal_type(expected)
37+
reveal_type(result)
38+
39+
40+
defdemo_args_iter_not_comparable_with_key()->None:
41+
args= [object(),object(),object()]
42+
key=id
43+
expected=max(args,key=id)
44+
result=my.max(args,key=key)
45+
print(args,key,expected,result,sep='\n')
46+
assertresult==expected
47+
ifTYPE_CHECKING:
48+
reveal_type(args)
49+
reveal_type(key)
50+
reveal_type(expected)
51+
reveal_type(result)
52+
53+
54+
defdemo_empty_iterable_with_default()->None:
55+
args:List[float]= []
56+
default=None
57+
expected=None
58+
result=my.max(args,default=default)
59+
print(args,default,expected,result,sep='\n')
60+
assertresult==expected
61+
ifTYPE_CHECKING:
62+
reveal_type(args)
63+
reveal_type(default)
64+
reveal_type(expected)
65+
reveal_type(result)
66+
67+
68+
defdemo_different_key_return_type()->None:
69+
args=iter('banana kiwi mango apple'.split())
70+
key=len
71+
expected='banana'
72+
result=my.max(args,key=key)
73+
print(args,key,expected,result,sep='\n')
74+
assertresult==expected
75+
ifTYPE_CHECKING:
76+
reveal_type(args)
77+
reveal_type(key)
78+
reveal_type(expected)
79+
reveal_type(result)
80+
81+
82+
defdemo_different_key_none()->None:
83+
args=iter('banana kiwi mango apple'.split())
84+
key=None
85+
expected='mango'
86+
result=my.max(args,key=key)
87+
print(args,key,expected,result,sep='\n')
88+
assertresult==expected
89+
ifTYPE_CHECKING:
90+
reveal_type(args)
91+
reveal_type(key)
92+
reveal_type(expected)
93+
reveal_type(result)
94+
95+
###################################### intentional type errors
96+
97+
deferror_reported_bug()->None:
98+
# example from https://github.com/python/typeshed/issues/4051
99+
top:Optional[int]=None
100+
try:
101+
my.max(5,top)
102+
exceptTypeErrorasexc:
103+
print(exc)
104+
105+
106+
deferror_args_iter_not_comparable()->None:
107+
try:
108+
my.max([None,None])
109+
exceptTypeErrorasexc:
110+
print(exc)
111+
112+
113+
deferror_single_arg_not_iterable()->None:
114+
try:
115+
my.max(1)
116+
exceptTypeErrorasexc:
117+
print(exc)
118+
119+
120+
defmain():
121+
forname,valinglobals().items():
122+
ifname.startswith('demo')orname.startswith('error'):
123+
print('_'*20,name)
124+
val()
125+
126+
if__name__=='__main__':
127+
main()
Lines changed: 96 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,96 @@
1+
fromtypingimportList,Callable,TypeVar
2+
3+
importpytest# type: ignore
4+
5+
importmymaxasmy
6+
7+
@pytest.fixture
8+
deffruits():
9+
return'banana kiwi mango apple'.split()
10+
11+
@pytest.mark.parametrize('args, expected', [
12+
([1,3],3),
13+
([3,1],3),
14+
([30,10,20],30),
15+
])
16+
deftest_max_args(args,expected):
17+
result=my.max(*args)
18+
assertresult==expected
19+
20+
@pytest.mark.parametrize('args, expected', [
21+
([1,3],1),
22+
([3,1],1),
23+
([30,10,20],10),
24+
])
25+
deftest_min_args(args,expected):
26+
result=my.min(*args)
27+
assertresult==expected
28+
29+
@pytest.mark.parametrize('iterable, expected', [
30+
([7],7),
31+
([1,3],3),
32+
([3,1],3),
33+
([30,10,20],30),
34+
])
35+
deftest_max_iterable(iterable,expected):
36+
result=my.max(iterable)
37+
assertresult==expected
38+
39+
40+
deftest_max_single_arg_not_iterable():
41+
msg="'int' object is not iterable"
42+
withpytest.raises(TypeError)asexc:
43+
my.max(1)
44+
assertexc.value.args[0]==msg
45+
46+
47+
deftest_max_empty_iterable_no_default():
48+
withpytest.raises(ValueError)asexc:
49+
my.max([])
50+
assertexc.value.args[0]==my.EMPTY_MSG
51+
52+
53+
@pytest.mark.parametrize('iterable, default, expected', [
54+
([7],-1,7),
55+
([],-1,-1),
56+
([],None,None),
57+
])
58+
deftest_max_empty_iterable_with_default(iterable,default,expected):
59+
result=my.max(iterable,default=default)
60+
assertresult==expected
61+
62+
63+
@pytest.mark.parametrize('key, expected', [
64+
(None,'mango'),
65+
(lambdax:x,'mango'),
66+
(len,'banana'),
67+
(lambdas:-len(s),'kiwi'),
68+
(lambdas:-ord(s[0]),'apple'),
69+
(lambdas:ord(s[-1]),'mango'),
70+
])
71+
deftest_max_iterable_with_key(
72+
fruits:List[str],
73+
key:Callable[[str],str],
74+
expected:str
75+
)->None:
76+
result=my.max(fruits,key=key)
77+
assertresult==expected
78+
79+
80+
'banana kiwi mango apple'
81+
82+
@pytest.mark.parametrize('key, expected', [
83+
(None,'apple'),
84+
(lambdax:x,'apple'),
85+
(len,'kiwi'),
86+
(lambdas:-len(s),'banana'),
87+
(lambdas:-ord(s[0]),'mango'),
88+
(lambdas:ord(s[-1]),'banana'),
89+
])
90+
deftest_min_iterable_with_key(
91+
fruits:List[str],
92+
key:Callable[[str],str],
93+
expected:str
94+
)->None:
95+
result=my.min(fruits,key=key)
96+
assertresult==expected

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp