ENH(sol,plot,.TC): PLOTter solutions

Author: ankostis

docs/source/plotting.rst

@@ -25,10 +25,17 @@ solution of the last computation, calling methods with arguments like this::
 
    The legend for all graphtik diagrams, generated by :func:`.legend()`.
 
-The same :meth:`.Plotter.plot()` method applies for :class:`.NetworkOperation`,
-:class:`.Network` & :class:`.ExecutionPlan`, each one capable to produce diagrams
-with increasing complexity.  Whenever possible, the top-level ``plot()`` methods
-delegates to the ones below.
+The same :meth:`.Plotter.plot()` method applies also for:
+
+- :class:`.NetworkOperation`
+- :class:`.Network`
+- :class:`.ExecutionPlan`
+- :class:`~.network.Solution`
+
+each one capable to producing diagrams with increasing complexity.
+Whenever possible, the top-level ``plot()`` methods will delegate to the ones below;
+specifically, the :term:`netop` keeps a transient reference to the last `plan`.
+BUT the `plan` does not hold such a reference, you have to plot the `solution`.
 
 For instance, when a net-operation has just been composed, plotting it will
 come out bare bone, with just the 2 types of nodes (data & operations), their

graphtik/network.py

@@ -63,7 +63,7 @@ def is_skip_evictions():
     return _execution_configs.get()["skip_evictions"]
 
 
-class Solution(ChainMap):
+class Solution(ChainMap, Plotter):
     """Collects outputs from operations, preserving :term:`overwrites`."""
 
     def __init__(self, plan, *args, **kw):
@@ -110,6 +110,12 @@ def overwrites(self) -> Mapping[Any, List]:
 
         return {k: v for k, v in dd.items() if len(v) > 1}
 
+    def _build_pydot(self, **kws):
+        """delegate to network"""
+        kws.setdefault("solution", self)
+        plotter = self.plan
+        return plotter._build_pydot(**kws)
+
 
 class _DataNode(str):
     """

test/test_plot.py

@@ -110,16 +110,32 @@ def test_plotters_hierarchy(pipeline: NetworkOperation, inputs, outputs):
     solution = pipeline.compute(inputs, outputs)
 
     # Plotting delegates to netwrok plan.
-    plan_dot = str(pipeline.plot(inputs=inputs, outputs=outputs))
+    netop_dot = str(pipeline.plot(inputs=inputs, outputs=outputs))
+    assert netop_dot
+    assert netop_dot != base_dot
+    assert pipeline.name in str(netop_dot)
+
+    # Plotting plan alone has not label.
+    plan_dot = str(pipeline.last_plan.plot(inputs=inputs, outputs=outputs))
     assert plan_dot
     assert plan_dot != base_dot
-    assert pipeline.name in str(plan_dot)
+    assert plan_dot != netop_dot
+    assert pipeline.name not in str(plan_dot)
 
     # Plot a plan + solution, which must be different from all before.
-    sol_plan_dot = str(pipeline.plot(inputs=inputs, outputs=outputs, solution=solution))
-    assert sol_plan_dot != base_dot
-    assert sol_plan_dot != plan_dot
-    assert pipeline.name in str(plan_dot)
+    sol_netop_dot = str(
+        pipeline.plot(inputs=inputs, outputs=outputs, solution=solution)
+    )
+    assert sol_netop_dot != base_dot
+    assert sol_netop_dot != netop_dot
+    assert sol_netop_dot != plan_dot
+    assert pipeline.name in str(netop_dot)
+
+    # Plot a solution, which must equal plan + sol.
+    sol_plan_dot = str(
+        pipeline.last_plan.plot(inputs=inputs, outputs=outputs, solution=solution)
+    )
+    assert str(solution.plot(inputs=inputs, outputs=outputs)) == sol_plan_dot
 
     plan = pipeline.last_plan
     pipeline.last_plan = None