test: refactor test utils

tests/utils/__init__.py

@@ -0,0 +1,27 @@
+from inspect import getsource
+from .type_conversion import type_conversion
+from .js import JS
+from .lua import Lua
+from .py import Py
+
+
+langs = [Py(), JS(), Lua()]
+
+def check_exprs(exprs):
+    expected_results = list(map(eval, exprs))
+    for lang in langs:
+        for i, e in enumerate(exprs):
+            checked_expr = lang.gen_expr(e)
+            fact_result = type_conversion(lang.eval(checked_expr))
+            lang.clear()
+            assert expected_results[i] == fact_result
+
+def check_func(func):
+    expected_results = func()
+    src = getsource(func)
+    for lang in langs:
+        target_src = lang.gen_func(src)
+        lang.load_func(target_src)
+        fact_result = type_conversion(lang.call_func(func.__name__))
+        lang.clear()
+        assert expected_results == fact_result

tests/utils/js.py

@@ -0,0 +1,26 @@
+import execjs
+from kithon import Transpiler, analogs
+
+
+class JS:
+    def __init__(self):
+        self.runtime = execjs.get()
+        self.generator = Transpiler(lang='js')
+
+    def load_func(self, func: str):
+        self.runtime = execjs.compile(func)
+
+    def call_func(self, name):
+        return self.eval(analogs.call(self.generator, name).render())
+
+    def eval(self, expr: str):
+        return self.runtime.eval(expr)
+
+    def gen_func(self, src):
+        return self.generator.generate(src, mode='block')
+
+    def gen_expr(self, src):
+        return self.generator.generate(src, mode='eval')
+
+    def clear(self):
+        self.generator.used.clear()

tests/utils/lua.py

@@ -0,0 +1,26 @@
+from lupa import LuaRuntime
+from kithon import Transpiler, analogs
+
+
+class Lua:
+    def __init__(self):
+        self.runtime = LuaRuntime(unpack_returned_tuples=True)
+        self.generator = Transpiler(lang='lua')
+
+    def load_func(self, func: str):
+        self.runtime.execute(func)
+
+    def call_func(self, name):
+        return self.eval(analogs.call(self.generator, name).render())
+
+    def eval(self, expr: str):
+        return self.runtime.eval(expr)
+
+    def gen_func(self, src):
+        return self.generator.generate(src, mode='block')
+
+    def gen_expr(self, src):
+        return self.generator.generate(src, mode='eval')
+
+    def clear(self):
+        self.generator.used.clear()

tests/utils/py.py

@@ -0,0 +1,26 @@
+from lupa import LuaRuntime
+from kithon import Transpiler, analogs
+
+
+class Py:
+    def __init__(self):
+        self.runtime = None
+        self.generator = Transpiler()
+
+    def load_func(self, func: str):
+        exec(func, globals())
+
+    def call_func(self, name):
+        return eval(analogs.call(self.generator, name).render())
+
+    def eval(self, expr: str):
+        return eval(expr)
+
+    def gen_func(self, src):
+        return self.generator.generate(src, mode='block')
+
+    def gen_expr(self, src):
+        return self.generator.generate(src, mode='eval')
+
+    def clear(self):
+        self.generator.used.clear()

tests/type_conversion.py → tests/utils/type_conversion.py

@@ -3,7 +3,7 @@
 
 lua_table = type(lupa.LuaRuntime().table())
 
-def type_conversion(data, lang=''):
+def type_conversion(data):
     if isinstance(data, lua_table):
         return list(map(type_conversion, data.values()))
     return data