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

 def test_getstatementrange_triple_quoted(self):
     #print str(self.source)
     source = Source("""hello('''
     ''')""")
     s = source.getstatement(0)
     assert s == str(source)
     s = source.getstatement(1)
     assert s == str(source)

def test_source_putaround_simple():
    source = Source("raise ValueError")
    source = source.putaround(
        "try:", """\
        except ValueError:
            x = 42
        else:
            x = 23""")
    assert str(source)=="""\

 def test_getstatementrange_bug(self):
     source = Source("""\
         try:
             x = (
                y +
                z)
         except:
             pass
     """)
     assert len(source) == 6
     assert source.getstatementrange(2) == (1, 4)

 def test_getstatementrange_bug2(self):
     source = Source("""\
         assert (
             33
             ==
             [
               X(3,
                   b=1, c=2
                ),
             ]
           )
     """)
     assert len(source) == 9
     assert source.getstatementrange(5) == (0, 9)

 def test_getstatementrange_bug2(self):
     py.test.skip("fix me (issue19)")
     source = Source("""\
         assert (
             33
             ==
             [
               X(3,
                   b=1, c=2
                ),
             ]
           )
     """)
     assert len(source) == 9
     assert source.getstatementrange(5) == (0, 9)

 def test_getstatementrange_within_constructs(self):
     source = Source("""\
         try:
             try: 
                 raise ValueError
             except SomeThing: 
                 pass 
         finally:
             42
     """)
     assert len(source) == 7
     assert source.getstatementrange(0) == (0, 7) 
     assert source.getstatementrange(1) == (1, 5) 
     assert source.getstatementrange(2) == (2, 3)
     assert source.getstatementrange(3) == (1, 5)
     assert source.getstatementrange(4) == (4, 5) 
     assert source.getstatementrange(5) == (0, 7)
     assert source.getstatementrange(6) == (6, 7) 

 def test_getstatementrange_within_constructs(self):
     source = Source("""\
         try:
             try:
                 raise ValueError
             except SomeThing:
                 pass
         finally:
             42
     """)
     assert len(source) == 7
     # check all lineno's that could occur in a traceback
     #assert source.getstatementrange(0) == (0, 7)
     #assert source.getstatementrange(1) == (1, 5)
     assert source.getstatementrange(2) == (2, 3)
     assert source.getstatementrange(3) == (3, 4)
     assert source.getstatementrange(4) == (4, 5)
     #assert source.getstatementrange(5) == (0, 7)
     assert source.getstatementrange(6) == (6, 7)

def test_source_strip_multiline():
    source = Source()
    source.lines = ["", " hello", "  "]
    source2 = source.strip()
    assert source2.lines == [" hello"]

def test_source_strips():
    source = Source("")
    assert source == Source()
    assert str(source) == ''
    assert source.strip() == source

 def test_compile_to_ast(self):
     import ast
     source = Source("x = 4")
     mod = source.compile(flag=ast.PyCF_ONLY_AST)
     assert isinstance(mod, ast.Module)
     compile(mod, "<filename>", "exec")

 def test_getstatementrange_with_syntaxerror_issue7(self):
     source = Source(":")
     py.test.raises(SyntaxError, lambda: source.getstatementrange(0))

 def test_getstatementrange_out_of_bounds_py3(self):
     source = Source("if xxx:\n   from .collections import something")
     r = source.getstatementrange(1)
     assert r == (1,2)