def test_declaration_with_comments(self):
declaration = ' # pre\n' \
' test mV = 10mV # in\n' \
' # post\n\n'
model = ModelParser.parse_block(declaration)
model_printer = ASTNestMLPrinter()
self.assertEqual(declaration, model_printer.print_node(model))
def test_function_call_with_comments(self):
function_call = ' # pre\n' \
' min(1,2) # in\n' \
' # post\n\n'
model = ModelParser.parse_block(function_call)
model_printer = ASTNestMLPrinter()
self.assertEqual(function_call, model_printer.print_node(model))
def test_assignment_with_comments(self):
assignment = ' # pre\n' \
' a = b # in\n' \
' # post\n' \
'\n'
model = ModelParser.parse_block(assignment)
model_printer = ASTNestMLPrinter()
self.assertEqual(assignment, model_printer.print_node(model))
def test_while_stmt_with_comments(self):
stmt = ' # pre\n' \
' while true: # in \n' \
' end\n' \
' # post\n\n'
model = ModelParser.parse_block(stmt)
model_printer = ASTNestMLPrinter()
self.assertEqual(stmt, model_printer.print_node(model))
def test_for_stmt_with_comments(self):
stmt = ' # pre\n' \
' for i in 10 - 3.14...10 + 3.14 step -1: # in\n' \
' end\n' \
' # post\n\n'
model = ModelParser.parse_block(stmt)
model_printer = ASTNestMLPrinter()
self.assertEqual(stmt, model_printer.print_node(model))
