Par exemple, lorsque vous souhaitez tester une entrée standard, il est difficile de saisir à chaque fois, mais la lecture d'un fichier externe ne fait qu'augmenter les déchets. Si vous souhaitez tester le code d'origine aussi proprement que possible, il semble préférable de remplacer input ().
Notez le contenu de l'entrée standard que vous souhaitez tester comme suit
stdin_test = """1 2
3 4
5 6
"""
Remplacer l'entrée () dans le générateur
def inp(t):
from io import StringIO
for i in StringIO(t):
yield i
input = inp(stdin_test).__next__
Exemple d'utilisation
a = [list(map(int, input().split())) for i in range(3)]
print(a)
>>> [[1, 2], [3, 4], [5, 6]]
Je ne pense pas qu'il y ait une meilleure façon fin
Recommended Posts