diff options
Diffstat (limited to 'bare-arm/main.c')
| -rw-r--r-- | bare-arm/main.c | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/bare-arm/main.c b/bare-arm/main.c index ca32dc459..6f5707afe 100644 --- a/bare-arm/main.c +++ b/bare-arm/main.c @@ -7,7 +7,7 @@ #include "py/runtime.h" #include "py/repl.h" -void do_str(const char *src) { +void do_str(const char *src, mp_parse_input_kind_t input_kind) { mp_lexer_t *lex = mp_lexer_new_from_str_len(MP_QSTR__lt_stdin_gt_, src, strlen(src), 0); if (lex == NULL) { return; @@ -16,7 +16,7 @@ void do_str(const char *src) { nlr_buf_t nlr; if (nlr_push(&nlr) == 0) { qstr source_name = lex->source_name; - mp_parse_node_t pn = mp_parse(lex, MP_PARSE_SINGLE_INPUT); + mp_parse_node_t pn = mp_parse(lex, input_kind); mp_obj_t module_fun = mp_compile(pn, source_name, MP_EMIT_OPT_NONE, true); mp_call_function_0(module_fun); nlr_pop(); @@ -28,7 +28,8 @@ void do_str(const char *src) { int main(int argc, char **argv) { mp_init(); - do_str("print('hello world!', list(x+1 for x in range(10)), end='eol\n')"); + do_str("print('hello world!', list(x+1 for x in range(10)), end='eol\\n')", MP_PARSE_SINGLE_INPUT); + do_str("for i in range(10):\n print(i)", MP_PARSE_FILE_INPUT); mp_deinit(); return 0; } |