【python】一回で全ての標準入力を与える方法を教えてください。

Question

前提

現在pythonを学び始めて1ヵ月程度です。
GoogleColabを使っています。

atcoder等の問題を解いている際に、標準入力される数値や文字が多くてプログラムのテストをしたくても手作業では時間が掛かってしまうことがあります。

現在の私ではプログラムが動くか試したいとき、input()の数だけenterを押す必要があるので、一回で全ての標準入力を与える方法があれば知りたいです。

実現したいこと

【a,b,c,の自然数が一行ずつ与えられるので全てを足した数を出力しなさい。】という問題で作ったプログラムが動くかテストしたいとき

====
1
2
3
====
a=int(input())
b=int(input())
c=int(input())
print(a+b+c) #出力>6

のように====に挟まれた文字が標準入力として与えられる。のようなライブラリや方法があれば教えていただきたいです。

Answer 1

==== ではなく、''' ですが、以下の様にしてもよいかと思います。

import sys
import io

input_string ='''
1
2
3
'''.lstrip()
org_stdin = sys.stdin
sys.stdin = io.StringIO(input_string)

#
a=int(input())
b=int(input())
c=int(input())
sys.stdin = org_stdin

print(a+b+c) #出力>6

Comments

[tamakoma]

回答ありがとうございます！
GoogleColab上では作動しなかったので、これを機にVSCodeに乗り換えました。
VSCode上では正常に動きました。助かりました。

Answer 2

colabのセル上で実行したい場合は、少し面倒くさいですが以下のようなコードでできなくもないです。

LINES = """
1
2
3
"""

このセルも毎回実行する必要あり

def inp():
    for line in LINES.split('\n'):
        yield line
g = inp()
next(g) # 先頭の改行をスキップ
def input():
    return next(g)

a=int(input())
b=int(input())
c=int(input())
print(a+b+c) #出力>6

Comments

[tamakoma]

質問の内容的に、Colab上でも作動したこちらの回答をベストアンサーにさせていただきました。
ありがとうございます。

Answer 3

コマンドのパイプで渡しみてはいかがでしょうか

$ echo "1\n2\n3" | python example.py
6