def test_traverse():
t = TableSymbol('t', '{name: string, amount: int}')
assert t in list(t._traverse())
expr = t.amount.sum()
trav = list(expr._traverse())
assert builtins.any(t.amount.isidentical(x) for x in trav)
def test_transform():
t = symbol('t', '5 * {x: int, y: int}')
expr = transform(t, z=t.x + t.y)
assert expr.fields == ['x', 'y', 'z']
assert builtins.any(
(t.x + t.y).isidentical(node) for node in expr._subterms())
def test_traverse():
t = symbol('t', 'var * {name: string, amount: int}')
assert t in list(t._traverse())
expr = t.amount.sum()
trav = list(expr._traverse())
assert builtins.any(t.amount.isidentical(x) for x in trav)
def test_transform():
t = symbol('t', '5 * {x: int, y: int}')
expr = transform(t, z=t.x + t.y)
assert expr.fields == ['x', 'y', 'z']
assert builtins.any((t.x + t.y).isidentical(node)
for node in expr._subterms())
def test_path_issue():
t = TableSymbol('t', "{topic: string, word: string, result: ?float64}")
t2 = transform(t,
sizes=t.result.map(lambda x: (x - MIN) * 10 / (MAX - MIN),
schema='float64',
name='size'))
assert builtins.any(t2.sizes.isidentical(node) for node in t2.children)
def test_path_issue():
t = symbol('t', "{topic: string, word: string, result: ?float64}")
t2 = transform(t, sizes=t.result.map(lambda x: (x - MIN)*10/(MAX - MIN),
schema='float64', name='size'))
assert builtins.any(t2.sizes.isidentical(node) for node in t2.children)
