def check_letter(num, letra):
return letra == 'TRWAGMYFPDXBNJZSQVHLCKE'[num%23]
def busca_dni(lista):
'''
>>> busca_dni(['12345678Z', '98765432M'])
'ok'
>>> busca_dni(['44556677A', '98765432M'])
'44556677A'
>>> busca_dni(['37485960P', '150150150', '13243546P'])
'150150150'
>>> busca_dni(['123456S', '13243546P'])
'123456S'
>>> busca_dni(['98765432M', '1234567890H'])
'1234567890H'
'''
for c in lista:
if len(c) != 9 or check_letter(int(c[0:len(c)-1]), c[-1]) == False:
return c
return'ok'
if __name__ =='__main__':
import doctest
print(doctest.testmod())