def test_fluent_function_kwarg():
d = DataFrame()
d1 = d.count(dude=22.0)
assert d1.filter is None
assert d1.child_expr is not None
assert ast.dump(d1.child_expr) == \
"Call(func=Attribute(value=ast_DataFrame(), attr='count', ctx=Load()), args=[], " \
"keywords=[keyword(arg='dude', value=Num(n=22.0))])"
def test_fluent_function_pos_arg():
d = DataFrame()
d1 = d.count(22.0)
assert d1.filter is None
assert d1.child_expr is not None
assert ast.dump(d1.child_expr) == \
"Call(func=Attribute(value=ast_DataFrame(), attr='count', ctx=Load()), " \
"args=[Num(n=22.0)], keywords=[])"
def test_render_func_with_df_arg():
d = DataFrame()
d1 = d.count(d)
expr, _ = render(d1)
assert isinstance(expr, ast.Call)
assert len(expr.args) == 1
arg1 = expr.args[0] # type: ast.AST
assert isinstance(arg1, ast_DataFrame)
assert arg1.dataframe is d
def test_render_func_with_args():
d = DataFrame()
d1 = d.count(10)
expr, _ = render(d1)
assert isinstance(expr, ast.Call)
assert len(expr.args) == 1
arg1 = expr.args[0]
assert isinstance(arg1, ast.Num)
assert arg1.n == 10
def test_render_base_call():
d = DataFrame()
d1 = d.count()
expr, _ = render(d1)
assert isinstance(expr, ast.Call)
assert len(expr.args) == 0
e_func = expr.func
assert isinstance(e_func, ast.Attribute)
e_val = e_func.value # type: ast.AST
assert isinstance(e_val, ast_DataFrame)
assert e_val.dataframe is d
