PythonMonkey/tests/python/test_functions.py at bc677bbda01a2cbdd030540e39de523a23a025c7 · Distributive-Network/PythonMonkey · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
import pythonmonkey as pm


def test_func_no_args():
  def f():
    return 42
  assert 42 == pm.eval("(f) => f()")(f)


def test_func_too_many_args():
  def f(a, b):
    return [a, b]
  assert [1, 2] == pm.eval("(f) => f(1, 2, 3)")(f)


def test_func_equal_args():
  def f(a, b):
    return [a, b]
  assert [1, 2] == pm.eval("(f) => f(1, 2)")(f)


def test_func_too_few_args():
  def f(a, b):
    return [a, b]
  assert [1, None] == pm.eval("(f) => f(1)")(f)


def test_default_func_no_args():
  def f(a, b, c=42, d=43):
    return [a, b, c, d]
  assert [None, None, 42, 43] == pm.eval("(f) => f()")(f)


def test_default_func_too_many_default_args():
  def f(a, b, c=42, d=43):
    return [a, b, c, d]
  assert [1, 2, 3, 4] == pm.eval("(f) => f(1, 2, 3, 4, 5)")(f)


def test_default_func_equal_default_args():
  def f(a, b, c=42, d=43):
    return [a, b, c, d]
  assert [1, 2, 3, 4] == pm.eval("(f) => f(1, 2, 3, 4)")(f)


def test_default_func_too_few_default_args():
  def f(a, b, c=42, d=43):
    return [a, b, c, d]
  assert [1, 2, 3, 43] == pm.eval("(f) => f(1, 2, 3)")(f)


def test_default_func_equal_args():
  def f(a, b, c=42, d=43):
    return [a, b, c, d]
  assert [1, 2, 42, 43] == pm.eval("(f) => f(1, 2)")(f)


def test_default_func_too_few_args():
  def f(a, b, c=42, d=43):
    return [a, b, c, d]
  assert [1, None, 42, 43] == pm.eval("(f) => f(1)")(f)


def test_vararg_func_no_args():
  def f(a, b, *args):
    return [a, b, *args]
  assert [None, None] == pm.eval("(f) => f()")(f)


def test_vararg_func_too_many_args():
  def f(a, b, *args):
    return [a, b, *args]
  assert [1, 2, 3] == pm.eval("(f) => f(1, 2, 3)")(f)


def test_vararg_func_equal_args():
  def f(a, b, *args):
    return [a, b, *args]
  assert [1, 2] == pm.eval("(f) => f(1, 2)")(f)


def test_vararg_func_too_few_args():
  def f(a, b, *args):
    return [a, b, *args]
  assert [1, None] == pm.eval("(f) => f(1)")(f)


def test_default_vararg_func_no_args():
  def f(a, b, c=42, d=43, *args):
    return [a, b, c, d, *args]
  assert [None, None, 42, 43] == pm.eval("(f) => f()")(f)


def test_default_vararg_func_too_many_default_args():
  def f(a, b, c=42, d=43, *args):
    return [a, b, c, d, *args]
  assert [1, 2, 3, 4, 5] == pm.eval("(f) => f(1, 2, 3, 4, 5)")(f)


def test_default_vararg_func_equal_default_args():
  def f(a, b, c=42, d=43, *args):
    return [a, b, c, d, *args]
  assert [1, 2, 3, 4] == pm.eval("(f) => f(1, 2, 3, 4)")(f)


def test_default_vararg_func_too_few_default_args():
  def f(a, b, c=42, d=43, *args):
    return [a, b, c, d, *args]
  assert [1, 2, 3, 43] == pm.eval("(f) => f(1, 2, 3)")(f)


def test_default_vararg_func_equal_args():
  def f(a, b, c=42, d=43, *args):
    return [a, b, c, d, *args]
  assert [1, 2, 42, 43] == pm.eval("(f) => f(1, 2)")(f)


def test_default_vararg_func_too_few_args():
  def f(a, b, c=42, d=43, *args):
    return [a, b, c, d, *args]
  assert [1, None, 42, 43] == pm.eval("(f) => f(1)")(f)

def test_kwargs_in_js():
  assert [1, 2, {"i": 42, "j": 43}] == pm.eval("(first, second, kwargs) => {return [first, second, kwargs]}")(1, 2, i=42, j=43)

def test_kwargs_in_js_arguments_array():
  # Note: arguments object is only available in non-arrow functions
  assert [1, 2, {"i": 42, "j": 43}] == pm.eval("(function() {return [...arguments]})")(1, 2, i=42, j=43)

def test_kwargs_pass_through():
  def py_func(*args, **kwargs):
    return [*args, kwargs];

  # Note: kwargs passed py1->js->py2 end up in args for py2 rather than kwargs, since JS has no equivalent to kwargs
  assert [1, 2, {"i": 42, "j": 43}, {}] == pm.eval("(py_func, ...args) => { return py_func(...args)}")(py_func, 1, 2, i=42, j=43)