def test_fixSemicolons_02():
""" don't put semicolons after comments """
src = '''
def afunc():
""" this
is
a
comment """
doStuff()
x = "this is not a comment"
'but this is'
'''
nodes = parseSource( src )
fixSemicolons( nodes )
# dumpTree( nodes )
# dumpNodes( nodes )
assert nodesToLines( nodes ) == [
"def afunc():",
' """ this',
" is",
" a",
' comment """',
" doStuff();",
" x = \"this is not a comment\";",
" 'but this is'",
]
def processFunction( func_node, in_class=False ):
fixIndents( func_node )
func_gath = FunctionConverter( in_class )
func_matches = func_gath.gather( func_node )
func_gath.processAll( func_matches )
gath_classes = [
AssignmentConverter,
BoolOpsConverter,
ComparisonConverter,
DecoratorConverter,
DictComprehensionConverter,
ExceptionConverter,
ForLoopConverter,
IfElifElseConverter,
KeyWordCallConverter,
ListComprehensionConverter,
ListSliceConverter,
NegateConverter,
SelfConverter,
StringInterpolationConverter,
SuperConverter,
TryExceptConverter,
TupleConverter,
WhileLoopConverter,
]
ConverterChain( gath_classes ).convertAll( func_node )
fixComments( func_node )
fixPassStatements( func_node )
fixSemicolons( func_node )
fixSimpleRenames( func_node )
def processModule( module_nodes ):
infos = AnonObj( class_names=[], func_names=[] )
ConverterChain( [ ImportNameConverter, ImportFromConverter ] ).convertAll( module_nodes )
class_gath = ClassConverter()
class_matches = class_gath.gather( module_nodes )
if not class_matches:
print( "NO CLASSES" )
else:
for class_match in class_matches:
print( "\nCLASS", class_match.name )
class_gath.processOne( class_match )
fixComments( class_match.suite )
func_matches = FunctionConverter( in_class=True ).gather( class_match.suite )
for func_match in func_matches:
print( "----METHOD", func_match.name )
processFunction( func_match.node, in_class=True )
infos.class_names.append( str( class_match.name ).strip() )
func_matches = FunctionConverter().gather( module_nodes )
for func_match in func_matches:
print( "FUNCTION ", func_match.name )
processFunction( func_match.node )
infos.func_names.append( str( func_match.name ).strip() )
fixComments( module_nodes )
fixPassStatements( module_nodes )
fixSemicolons( module_nodes )
fixSimpleRenames( module_nodes )
return infos
def test_fixSemicolons_01():
src = """
x = 1
callFunc()
y = callFunc()
a, b, c = \
1, 2, 3
"""
nodes = parseSource( src )
fixSemicolons( nodes )
# dumpTree( nodes )
# dumpNodes( nodes )
assert nodesToLines( nodes ) == [
"x = 1;",
"callFunc();",
"y = callFunc();",
"a, b, c = 1, 2, 3;",
]
def processTestModule(module_nodes):
infos = AnonObj(class_names=[], func_names=[])
# convert asserts before anything else
assert_cvtr = AssertToChaiExpectConverter()
assert_cvtr.processAll(assert_cvtr.gather(module_nodes))
ConverterChain([ImportNameConverter,
ImportFromConverter]).convertAll(module_nodes)
class_gath = UnittestClassToMochaDescribeConverter("unittest.TestCase")
class_matches = class_gath.gather(module_nodes)
if not class_matches:
print("NO CLASSES")
else:
for class_match in class_matches:
print("\nCLASS", class_match.name)
class_gath.processOne(class_match)
fixComments(class_match.suite)
func_matches = PytestMethodToMochaItConverter(
in_class=True).gather(class_match.suite)
for func_match in func_matches:
print("----METHOD", func_match.name)
processTestFunction(func_match.node, in_class=True)
infos.class_names.append(class_match.name.toString())
func_matches = PytestMethodToMochaItConverter().gather(module_nodes)
for func_match in func_matches:
print("FUNCTION ", func_match.name)
processTestFunction(func_match.node)
infos.func_names.append(func_match.name.toString())
fixComments(module_nodes)
fixPassStatements(module_nodes)
fixSimpleRenames(module_nodes)
fixSemicolons(module_nodes)
return infos