def test_getsTabwidthForFunctionAtRootScope(self):
src = trimLines("""
def myFn(self):
pass
""")
sourcenode = writeSourceAndCreateNode(src)
em = ExtractMethod(sourcenode, coords(2, 0), coords(2, 9), "foobah")
self.assertEqual(em.getTabwidthOfParentFunction(), 0)
def test_getsTabwidthForFunctionAtRootScope(self):
src=trimLines("""
def myFn(self):
pass
""")
sourcenode = writeSourceAndCreateNode(src)
em = ExtractMethod(sourcenode, coords(2, 0),
coords(2, 9), "foobah")
self.assertEqual(em.getTabwidthOfParentFunction(), 0)
def test_getsTabwidthForSimpleMethod(self):
src = trimLines("""
class MyClass:
def myMethod(self):
pass
""")
sourcenode = writeSourceAndCreateNode(src)
em = ExtractMethod(sourcenode, coords(3, 11), coords(3, 17), "foobah")
self.assertEqual(em.getTabwidthOfParentFunction(), 4)
def test_getsTabwidthForSimpleMethod(self):
src=trimLines("""
class MyClass:
def myMethod(self):
pass
""")
sourcenode = writeSourceAndCreateNode(src)
em = ExtractMethod(sourcenode, coords(3, 11),
coords(3, 17), "foobah")
self.assertEqual(em.getTabwidthOfParentFunction(), 4)
