Added assert and schema test cases

Signed-off-by: Vishal <[email protected]>

test/corpus/assert.txt

@@ -0,0 +1,46 @@
+================================================================================
+assert statements
+================================================================================
+
+assert x > 1
+
+--------------------------------------------------------------------------------
+
+(module
+    (assert_statement
+    (comparison_operator
+        (identifier)
+        (integer))))
+
+================================================================================
+assert statements using if expression
+================================================================================
+
+assert a > 2 if a
+
+--------------------------------------------------------------------------------
+
+(module
+    (assert_statement
+    (comparison_operator
+        (identifier)
+        (integer))
+    (identifier)))
+
+================================================================================
+assert statements with error message
+================================================================================
+
+assert a == b, "SOS"
+
+--------------------------------------------------------------------------------
+
+(module
+      (assert_statement
+        (comparison_operator
+          (identifier)
+          (identifier))
+        (string
+          (string_start)
+          (string_content)
+          (string_end))))

test/corpus/schema.txt

@@ -0,0 +1,22 @@
+================================================================================
+Schema Statement
+================================================================================
+
+schema Person:
+    firstName: str = "John"
+
+--------------------------------------------------------------------------------
+
+(module
+    (schema_statement
+    (identifier)
+    (block
+        (assignment
+        (dotted_name
+            (identifier))
+        (type
+            (basic_type))
+        (string
+            (string_start)
+            (string_content)
+            (string_end))))))

test/corpus/stmt.txt

@@ -16,7 +16,7 @@ if True: a = 1
         (integer)))))
 
 ================================================================================
-if statement inline
+if statement Basic
 ================================================================================
 
 if True:
@@ -37,3 +37,64 @@ if True:
         (dotted_name
           (identifier))
         (integer)))))
+
+================================================================================
+if statement with else
+================================================================================
+
+if True: 
+  a = 1 
+else: 
+  b = 2
+
+--------------------------------------------------------------------------------
+
+(module
+  (if_statement
+    (true)
+    (block
+      (assignment
+        (dotted_name
+          (identifier))
+        (integer)))
+    (else_clause
+      (block
+        (assignment
+          (dotted_name
+            (identifier))
+          (integer))))))
+
+================================================================================
+if statement with elif and else
+================================================================================
+
+if True:
+  a = 1
+elif False:
+  b = 2
+else:
+  c = 3
+
+--------------------------------------------------------------------------------
+
+(module
+  (if_statement
+    (true)
+    (block
+      (assignment
+        (dotted_name
+          (identifier))
+        (integer)))
+    (elif_clause
+      (false)
+      (block
+        (assignment
+          (dotted_name
+            (identifier))
+          (integer))))
+    (else_clause
+      (block
+        (assignment
+          (dotted_name
+            (identifier))
+          (integer))))))