bash-like heredoc for data input

Question

Sometimes it would be handy to test a script that reads data from a file using inline data (so that both the data and the code would be in the same file). In bash this can be done with a heredoc:

while read l;do
    echo $l
done << EOF
test
test2
test3
EOF

In some real code there would of course happen something more apart from writing out the lines. Suppose I would do something similar in python:

def read_file(f):
    for line in f.readlines():
        print(line.replace('\n',''))

with open('input.txt') as f:
    read_file(f)

What would be the best way to provide the content of input.txt to read_file() inline?

hek2mgl · Accepted Answer · 2017-07-31 11:34:47Z

4

You can use StringIO:

from io import StringIO    

f = StringIO('''\
foo
bar
test
''')

read_file(f)

The above code works with Python3. In Python2 use:

from StringIO import StringIO

edited Jul 31, 2017 at 11:34

answered Jul 31, 2017 at 11:29

hek2mgl

159k31 gold badges263 silver badges279 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

bash-like heredoc for data input

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related