(df, client, tmpdir)
| 29 | |
| 30 | @pytest.mark.filterwarnings("ignore:DataFrameGroupBy.apply operated ") |
| 31 | def test_analyze(df, client, tmpdir): |
| 32 | pytest.importorskip("crick") |
| 33 | filename = str(tmpdir / "analyze") |
| 34 | expr = df.groupby(df.columns[1]).apply( |
| 35 | lambda x: x, meta=df._meta.groupby(df.columns[1]).apply(lambda x: x) |
| 36 | ) |
| 37 | digraph = expr.analyze(filename=filename) |
| 38 | assert os.path.exists(filename + ".svg") |
| 39 | for exp in expr.optimize().walk(): |
| 40 | assert any(exp._name in el for el in digraph.body) |