def test_bogus_function_call(): df1 = DataFrame(expr=ast.Call(func=ast.BinOp())) with pytest.raises(Exception) as e: dumps(df1) assert "BinOp" in str(e.value)
def test_nested_dataframe(): df1 = DataFrame() df2 = DataFrame(expr=ast_DataFrame(df1)) r = dumps(df1 + df2) assert '\n'.join(r) == '''df_1 = DataFrame()
def test_python_builtin_function(): df = DataFrame() df1 = abs(df.x) r = dumps(df1) assert '\n'.join(r) == '''df_1 = DataFrame()
def test_callback(): df = DataFrame() df1 = df.map(lambda e: e.x) r = dumps(df1) assert '\n'.join(r) == '''df_1 = DataFrame()
def test_filter(): df = DataFrame() df1 = df[df.x > 0] r = dumps(df1) assert '\n'.join(r) == '''df_1 = DataFrame()
def test_binary_logic_operator(operator_text, op): df = DataFrame() df1 = op(df.x > 1, df.y < 1) r = dumps(df1) assert '\n'.join(r) == f'''df_1 = DataFrame()
def test_python_kw_args(): import numpy df = DataFrame() df1 = numpy.histogram(df.x, bins=50) r = dumps(df1) # type: ignore assert '\n'.join(r) == '''df_1 = DataFrame()
def test_python_np_function(): import numpy df = DataFrame() df1 = numpy.histogram(df.x) r = dumps(df1) # type: ignore assert '\n'.join(r) == '''df_1 = DataFrame()
def test_python_other_function(): @user_func def doit(i: float): assert 'do not call this' df = DataFrame() df1 = doit(df.x) r = dumps(df1) assert '\n'.join(r) == '''df_1 = DataFrame()