From e5677cb9a670ca9e1e4c5d47f3c130622c619772 Mon Sep 17 00:00:00 2001
From: Bruce Hill <bruce@bruce-hill.com>
Date: Tue, 26 Aug 2025 15:51:52 -0400
Subject: Add formatting for asserts

---
 src/formatter/formatter.c | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src/formatter')

diff --git a/src/formatter/formatter.c b/src/formatter/formatter.c
index 5dd65d04..6f0fc5aa 100644
--- a/src/formatter/formatter.c
+++ b/src/formatter/formatter.c
@@ -47,6 +47,13 @@ OptionalText_t format_inline_code(ast_t *ast, Table_t comments) {
     /*inline*/ case ConvertDef:
     /*inline*/ case DocTest:
         return NONE_TEXT;
+    /*inline*/ case Assert: {
+        DeclareMatch(assert, ast, Assert);
+        Text_t expr = fmt_inline(assert->expr, comments);
+        if (!assert->message) return Texts("assert ", expr);
+        Text_t message = fmt_inline(assert->message, comments);
+        return Texts("assert ", expr, ", ", message);
+    }
     /*inline*/ case Lambda: {
         DeclareMatch(lambda, ast, Lambda);
         Text_t code = Texts("func(", format_inline_args(lambda->args, comments));
@@ -662,6 +669,13 @@ Text_t format_code(ast_t *ast, Table_t comments, Text_t indent) {
         }
         return code;
     }
+    /*multiline*/ case Assert: {
+        DeclareMatch(assert, ast, Assert);
+        Text_t expr = fmt(assert->expr, comments, indent);
+        if (!assert->message) return Texts("assert ", expr);
+        Text_t message = fmt(assert->message, comments, indent);
+        return Texts("assert ", expr, ", ", message);
+    }
     /*multiline*/ case BINOP_CASES: {
         if (inlined_fits) return inlined;
         binary_operands_t operands = BINARY_OPERANDS(ast);
-- 
cgit v1.2.3