js_snippets.py

import unittest

from macropy.experimental.js_snippets import macros, pyjs, js, std_lib_script
from macropy.tracing import macros, require


from selenium import webdriver

class Tests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
    @classmethod
    def tearDownClass(cls):
        cls.driver.close()

    def exec_js(self, script):
        return Tests.driver.execute_script(
            std_lib_script + "return " + script
        )

    def exec_js_func(self, script, *args):
        arg_list = ", ".join("arguments[%s]" % i for i in range(len(args)))
        return Tests.driver.execute_script(
            std_lib_script + "return (" + script + ")(%s)" % arg_list,
            *args
        )
    def test_literals(self):
        # these work
        with require:
            self.exec_js(js[10]) == 10
            self.exec_js(js["i am a cow"]) == "i am a cow"

        # these literals are buggy, and it seems to be PJs' fault
        # ??? all the results seem to turn into strings ???
        with require:
            self.exec_js(js[3.14]) == str(3.14)
            self.exec_js(js[[1, 2, 'lol']]) == str([1, 2, 'lol'])
            self.exec_js(js[{"moo": 2, "cow": 1}]) == str({"moo": 2, "cow": 1})

        # set literals don't work so this throws an exception at macro-expansion time
        #self.exec_js(js%{1, 2, 'lol'})

    def test_executions(self):
        with require:
            self.exec_js(js[(lambda x: x * 2)(10)]) == 20
            self.exec_js(js[sum([x for x in range(10) if x > 5])]) == 30

    def test_pyjs(self):
        # cross-compiling a trivial predicate
        code, javascript = pyjs[lambda x: x > 5 and x % 2 == 0]

        for i in range(10):
            with require:
                code(i) == self.exec_js_func(javascript, i)


        code, javascript = pyjs[lambda n: [
            x for x in range(n)
            if 0 == len([
                y for y in range(2, x-2)
                if x % y == 0
            ])
        ]]
        # this is also wrongly stringifying the result =(
        with require:
            str(code(20)) == str(self.exec_js_func(javascript, 20))