79 lines
3.4 KiB
Python
79 lines
3.4 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from youtube_dl.jsinterp2.jsgrammar import Token
|
|
from youtube_dl.jsinterp2.tstream import _ASSIGN_OPERATORS
|
|
|
|
tests = [
|
|
{'code': 'var x = [1,2,3]; x[0] = 4; x[0] = 5; x[2] = 7; return x;',
|
|
'asserts': [{'value': [5, 2, 7]}],
|
|
'ast': [(Token.VAR,
|
|
zip(['x'],
|
|
[(Token.ASSIGN,
|
|
None,
|
|
(Token.OPEXPR, [
|
|
(Token.MEMBER, (Token.ARRAY, [
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
|
(Token.MEMBER, (Token.INT, 1), None, None)]), None),
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
|
(Token.MEMBER, (Token.INT, 2), None, None)]), None),
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [
|
|
(Token.MEMBER, (Token.INT, 3), None, None)]), None)
|
|
]), None, None),
|
|
]),
|
|
None)
|
|
])
|
|
),
|
|
(Token.EXPR, [
|
|
(Token.ASSIGN,
|
|
_ASSIGN_OPERATORS['='][1],
|
|
(Token.OPEXPR, [
|
|
(Token.MEMBER, (Token.ID, 'x'),
|
|
None,
|
|
(Token.ELEM,
|
|
(Token.EXPR, [
|
|
(Token.ASSIGN,
|
|
None,
|
|
(Token.OPEXPR, [(Token.MEMBER, (Token.INT, 0), None, None)]),
|
|
None)
|
|
]),
|
|
None))
|
|
]),
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.INT, 4), None, None)]), None)
|
|
)
|
|
]),
|
|
(Token.EXPR, [
|
|
(Token.ASSIGN,
|
|
_ASSIGN_OPERATORS['='][1],
|
|
(Token.OPEXPR, [(Token.MEMBER, (Token.ID, 'x'),
|
|
None,
|
|
(Token.ELEM, (Token.EXPR, [
|
|
(Token.ASSIGN,
|
|
None,
|
|
(Token.OPEXPR, [(Token.MEMBER, (Token.INT, 0), None, None)]),
|
|
None)
|
|
]), None))
|
|
]),
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.INT, 5), None, None)]), None))
|
|
]),
|
|
(Token.EXPR, [
|
|
(Token.ASSIGN,
|
|
_ASSIGN_OPERATORS['='][1],
|
|
(Token.OPEXPR, [(Token.MEMBER, (Token.ID, 'x'),
|
|
None,
|
|
(Token.ELEM, (Token.EXPR, [
|
|
(Token.ASSIGN,
|
|
None,
|
|
(Token.OPEXPR, [(Token.MEMBER, (Token.INT, 2), None, None)]),
|
|
None)
|
|
]), None))
|
|
]),
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.INT, 7), None, None)]), None))
|
|
]),
|
|
(Token.RETURN,
|
|
(Token.EXPR, [
|
|
(Token.ASSIGN, None, (Token.OPEXPR, [(Token.MEMBER, (Token.ID, 'x'), None, None)]), None)
|
|
])
|
|
)]
|
|
}
|
|
]
|