added script to generate random noja code

tests/randomizer.py

@@ -0,0 +1,201 @@
+
+from random import randint
+
+NODE_KINDS = ['expr', 'if-else', 'while', 'do-while', 'return', 'comp', 'func']
+EXPR_KINDS = ['int', 'float', 'string', 'array', 'object']
+
+def generate_ident():
+	return 'Apple'
+
+def generate_string_expr_node():
+	return 'Hello, world!'
+
+def generate_node(kind=None, depth=0):
+
+	if kind == None and depth > 3:
+		return {
+			'kind': 'comp',
+			'body': []
+		}
+
+	if kind == None:
+		kind = NODE_KINDS[randint(0, len(NODE_KINDS)-1)]
+
+	assert kind in NODE_KINDS
+
+	if kind == 'expr':
+	
+		return generate_expr_node(None, depth+1)
+
+	elif kind == 'if-else':
+		
+		branches = randint(1, 2)
+		assert branches in [1, 2]
+
+		if branches == 1:
+			return {
+				'kind': 'if-else', 
+				'cond': generate_expr_node(None, depth+1),
+				'if-case': generate_node(None, depth+1)
+			}
+		else:
+			return {
+				'kind': 'if-else', 
+				'cond': generate_expr_node(None, depth+1),
+				'if-case': generate_node(None, depth+1), 
+				'else-case': generate_node(None, depth+1)
+			}
+
+	elif kind == 'while':
+		
+		return {
+			'kind': 'while',
+			'cond': generate_expr_node(None, depth+1),
+			'body': generate_node(None, depth+1)
+		}
+
+	elif kind == 'do-while':
+		
+		return {
+			'kind': 'do-while',
+			'cond': generate_expr_node(None, depth+1),
+			'body': generate_node(None, depth+1)
+		}
+
+	elif kind == 'return':
+		
+		return {
+			'kind': 'return',
+			'value': generate_expr_node(None, depth+1)
+		}
+
+	elif kind == 'comp':
+
+		n = randint(0, 3)
+
+		return {
+			'kind': 'comp',
+			'body': [generate_node(None, depth+1) for i in range(n)]
+		}
+
+	elif kind == 'func':
+		
+		argc = randint(0, 3)
+
+		return {
+			'kind': 'func',
+			'name': generate_ident(),
+			'args': [generate_ident() for i in range(argc)],
+			'body': generate_node(None, depth+1)
+		}
+
+def generate_expr_node(expr_kind=None, depth=0):
+
+	if expr_kind == None:
+		expr_kind = EXPR_KINDS[randint(0, len(EXPR_KINDS)-1)]
+
+	assert expr_kind in EXPR_KINDS
+
+	if depth > 3:
+		expr_kind = ['int', 'float', 'string'][randint(0, 2)]
+
+	if expr_kind == 'int':
+
+		return {
+			'kind': 'expr',
+			'expr-kind': 'int',
+			'value': randint(0, 100)
+		}
+
+	elif expr_kind == 'float':
+
+		return {
+			'kind': 'expr',
+			'expr-kind': 'float',
+			'value': float(str(randint(0, 100)) + '.' + str(randint(0, 100)))
+		}
+
+	elif expr_kind == 'string':
+
+		return {
+			'kind': 'expr',
+			'expr-kind': 'string',
+			'value': generate_string_expr_node()
+		}
+
+	elif expr_kind == 'array':
+
+		n = randint(0, 3)
+
+		return {
+			'kind': 'expr',
+			'expr-kind': 'array',
+			'items': [generate_expr_node(None, depth+1) for i in range(n)]
+		}
+
+	elif expr_kind == 'object':
+
+		n = randint(0, 3)
+
+		return {
+			'kind': 'expr',
+			'expr-kind': 'object',
+			'items': [(generate_expr_node(None, depth+1), generate_expr_node(None, depth+1)) for i in range(n)]
+		}
+
+def serialize_expr(node):
+
+	if node['expr-kind'] == 'int':
+			
+		return str(node['value'])
+
+	elif node['expr-kind'] == 'float':
+		
+		return str(node['value'])
+
+	elif node['expr-kind'] == 'string':
+		
+		return '"' + node['value'] + '"'
+
+	elif node['expr-kind'] == 'array':
+		
+		return '[' + ', '.join([serialize_expr(item) for item in node['items']]) + ']'
+
+	elif node['expr-kind'] == 'object':
+		
+		return '{' + ', '.join([serialize_expr(item[0]) + ': ' + serialize_expr(item[1]) for item in node['items']]) + '}'
+
+def serialize(node):
+
+	if node['kind'] == 'expr':
+
+		return serialize_expr(node) + ';'
+
+	elif node['kind'] == 'if-else':
+
+		if 'else-case' in node:
+			return 'if ' + serialize_expr(node['cond']) + ': ' + serialize(node['if-case']) + ' else ' + serialize(node['else-case'])
+		else:
+			return 'if ' + serialize_expr(node['cond']) + ': ' + serialize(node['if-case'])
+
+	elif node['kind'] == 'while':
+
+		return 'while ' + serialize_expr(node['cond']) + ': ' + serialize(node['body'])
+
+	elif node['kind'] == 'do-while':
+
+		return 'do ' + serialize(node['body']) + ' while: ' + serialize_expr(node['cond']) + ';'
+
+	elif node['kind'] == 'return':
+
+		return 'return ' + serialize_expr(node['value']) + ';'
+
+	elif node['kind'] == 'comp': 
+
+		return '{' + ''.join([serialize(stmt) for stmt in node['body']]) + '}'
+
+	elif node['kind'] == 'func':
+
+		return 'fun ' + node['name'] + '(' + ', '.join(node['args']) + ') ' + serialize(node['body'])
+
+print(serialize(generate_node()))