Bruce/tomo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Activity

Fix doctests for declarations

Browse Source
This commit is contained in:
Bruce Hill 2024-02-12 02:07:12 -05:00
parent ade25eeb5a
commit da1081387f
2 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
compile.c
View File

@ -306,11 +306,23 @@ CORD compile(ast_t *ast)
case DocTest: {
auto test = Match(ast, DocTest);
CORD src = heap_strn(test->expr->start, (size_t)(test->expr->end - test->expr->start));
return CORD_asprintf(
"__test(%r, %r, %r);\n",
compile(WrapAST(test->expr, StringLiteral, .cord=src)),
compile(test->expr),
compile(WrapAST(test->expr, StringLiteral, .cord=test->output)));
if (test->expr->tag == Declare) {
auto decl = Match(test->expr, Declare);
return CORD_asprintf(
"__declare(%r, %r);\n__test(%r, %r, %r);\n",
compile(decl->var), compile(decl->value),
compile(WrapAST(test->expr, StringLiteral, .cord=src)),
compile(decl->var),
compile(WrapAST(test->expr, StringLiteral, .cord=test->output)));
} else if (test->expr->tag == Assign) {
errx(1, "Not implemented");
} else {
return CORD_asprintf(
"__test(%r, %r, %r);\n",
compile(WrapAST(test->expr, StringLiteral, .cord=src)),
compile(test->expr),
compile(WrapAST(test->expr, StringLiteral, .cord=test->output)));
}
}
case FieldAccess: {
auto f = Match(ast, FieldAccess);

2
nextlang.h
View File

@ -78,7 +78,7 @@ CORD as_cord(void *x, const char *fmt, ...);
#define say(str) puts(CORD_to_const_char_star(__cord(str)))
#define __test(src, expr, expected) do { \
CORD __result = __cord(expr); \
say(CORD_catn(5, USE_COLOR ? "\x1b[33;1m>>>\x1b[0m " : ">>> ", src, USE_COLOR ? "\n\x1b[0;2m=\x1b[0;35m " : "\n=== ", __result, "\x1b[m")); \
say(CORD_catn(5, USE_COLOR ? "\x1b[33;1m>>>\x1b[0m " : ">>> ", src, USE_COLOR ? "\n\x1b[0;2m=\x1b[0;35m " : "\n= ", __result, "\x1b[m")); \
if (expected && CORD_cmp(__result, expected)) { \
errx(1, "I expected:\n%s but got:\n%s", CORD_to_const_char_star(expected), CORD_to_const_char_star(__result)); \
} \