(fuse)
| 1498 | |
| 1499 | @pytest.mark.parametrize("fuse", [True, False]) |
| 1500 | def test_tree_repr(fuse): |
| 1501 | s = from_pandas(pd.Series(range(10))).expr.tree_repr() |
| 1502 | assert ("<pandas>" in s) or ("<series>" in s) |
| 1503 | |
| 1504 | df = timeseries() |
| 1505 | expr = ((df.x + 1).sum(skipna=False) + df.y.mean()).expr |
| 1506 | |
| 1507 | # Check result before optimization |
| 1508 | s = expr.tree_repr() |
| 1509 | assert "Sum:" in s |
| 1510 | assert "Add:" in s |
| 1511 | assert "Mean:" in s |
| 1512 | if pyarrow_strings_enabled(): |
| 1513 | assert "ArrowStringConversion" in s |
| 1514 | assert "AlignPartitions:" not in s |
| 1515 | if pyarrow_strings_enabled(): |
| 1516 | assert str(df.frame.seed) in s.lower() |
| 1517 | else: |
| 1518 | assert str(df.seed) in s.lower() |
| 1519 | |
| 1520 | # Check result after optimization |
| 1521 | optimized = expr.optimize(fuse=fuse) |
| 1522 | s = str(optimized.tree_repr()) |
| 1523 | assert "Sum(Chunk):" in s |
| 1524 | assert "Sum(TreeReduce): split_every=False" in s |
| 1525 | assert "Add:" in s |
| 1526 | assert "Mean:" not in s |
| 1527 | assert "AlignPartitions:" not in s |
| 1528 | assert "True" not in s |
| 1529 | assert "None" not in s |
| 1530 | assert "skipna=False" in s |
| 1531 | if pyarrow_strings_enabled(): |
| 1532 | assert str(df.frame.seed) in s.lower() |
| 1533 | else: |
| 1534 | assert str(df.seed) in s.lower() |
| 1535 | if fuse: |
| 1536 | assert "Fused" in s |
| 1537 | assert "|" in s |
| 1538 | |
| 1539 | |
| 1540 | def test_random_partitions(df, pdf): |
nothing calls this directly
no test coverage detected
searching dependent graphs…