| 1406 | |
| 1407 | @xfail_gpu("Cannot apply lambda function in cudf") |
| 1408 | def test_size_optimized(df): |
| 1409 | with pytest.warns(UserWarning, match="metadata"): |
| 1410 | expr = (df.x + 1).apply(lambda x: x).size |
| 1411 | out = optimize(expr) |
| 1412 | expected = optimize(df.x.size) |
| 1413 | assert out._name == expected._name |
| 1414 | |
| 1415 | with pytest.warns(UserWarning, match="metadata"): |
| 1416 | expr = (df + 1).apply(lambda x: x, axis=1).size |
| 1417 | out = optimize(expr) |
| 1418 | expected = optimize(df.size) |
| 1419 | assert out._name == expected._name |
| 1420 | |
| 1421 | |
| 1422 | def test_series_iter(df, pdf): |