def test_source_putaround():
source = Source()
source = source.putaround("""
if 1:
x=1
-> """)
assert str(source).strip() == "if 1:\n x=1"