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)