From eb6f40d25aa9b0be57f2ed49535054565eafc828 Mon Sep 17 00:00:00 2001
From: chrimaho <chrismahoney@hotmail.com>
Date: Mon, 5 Apr 2021 10:07:33 +1000
Subject: [PATCH] Add AssertionError

---
 src/parse/parse_parameters.ts | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/parse/parse_parameters.ts b/src/parse/parse_parameters.ts
index 776ed4d..dddb01c 100644
--- a/src/parse/parse_parameters.ts
+++ b/src/parse/parse_parameters.ts
@@ -133,9 +133,22 @@ function parseReturnFromDefinition(parameters: string[]): Returns | null {
 
 function parseExceptions(body: string[]): Exception[] {
     const exceptions: Exception[] = [];
-    const pattern = /(?<!#.*)raise\s+([\w.]+)/;
 
+    // Add "assert" exceptions
     for (const line of body) {
+        const pattern = /(?<!#.*)assert\s+/;
+        const match = line.match(pattern);
+
+        if (match == null) {
+            continue;
+        }
+
+        exceptions.push({ type: "AssertionError" });
+    }
+
+    // Add "raise" exceptions
+    for (const line of body) {
+        const pattern = /(?<!#.*)raise\s+([\w.]+)/;
         const match = line.match(pattern);
 
         if (match == null) {