Example #1
0
def test_plot_xyi_plot(node_counts_list, labels):
    """
    Test plotting xyi.
    """
    fig, ax, tax = parameters.plot_ternary_plot(counts_list=node_counts_list,
                                                labels=labels,
                                                is_nodes=True)
    assert isinstance(fig, matplotlib.figure.Figure)  # type: ignore
    assert isinstance(ax, matplotlib.axes.Axes)  # type: ignore
    assert isinstance(tax, TernaryAxesSubplot)
    plt.close("all")
Example #2
0
def test_plot_branch_plot(branch_counts_list, labels):
    """
    Test plotting branches.
    """
    fig, ax, tax = parameters.plot_ternary_plot(counts_list=branch_counts_list,
                                                labels=labels,
                                                is_nodes=False)
    assert isinstance(fig, matplotlib.figure.Figure)  # type: ignore
    assert isinstance(ax, matplotlib.axes.Axes)  # type: ignore
    assert isinstance(tax, ternary.ternary_axes_subplot.TernaryAxesSubplot)
    plt.close("all")
Example #3
0
 def plot_xyi(self, ):
     """
     Plot multi-network ternary XYI plot.
     """
     node_counts_list = [network.node_counts for network in self.networks]
     labels = [network.name for network in self.networks]
     return parameters.plot_ternary_plot(
         counts_list=node_counts_list,
         labels=labels,
         is_nodes=True,
     )
Example #4
0
 def plot_branch(
     self,
     label: Optional[str] = None
 ) -> Tuple[Figure, Axes, TernaryAxesSubplot]:
     """
     Plot ternary plot of branch types.
     """
     if label is None:
         label = self.name
     return plot_ternary_plot(counts_list=[self.branch_counts],
                              labels=[label],
                              is_nodes=False)
Example #5
0
def test_plot_branch_plot(branch_counts_list, labels, colors):
    """
    Test plotting branches.
    """
    fig, ax, tax = parameters.plot_ternary_plot(counts_list=branch_counts_list,
                                                labels=labels,
                                                is_nodes=False,
                                                colors=colors)
    assert isinstance(fig, Figure)
    assert isinstance(ax, Axes)
    assert isinstance(tax, TernaryAxesSubplot)
    plt.close("all")
Example #6
0
 def plot_branch(self, ):
     """
     Plot multi-network ternary branch type plot.
     """
     branch_counts_list = [
         network.branch_counts for network in self.networks
     ]
     labels = [network.name for network in self.networks]
     return parameters.plot_ternary_plot(
         counts_list=branch_counts_list,
         labels=labels,
         is_nodes=False,
     )
Example #7
0
 def plot_xyi(
     self,
     colors: Optional[List[Optional[str]]] = None,
 ):
     """
     Plot multi-network ternary XYI plot.
     """
     node_counts_list = [network.node_counts for network in self.networks]
     labels = [network.name for network in self.networks]
     return parameters.plot_ternary_plot(
         counts_list=node_counts_list,
         labels=labels,
         is_nodes=True,
         colors=colors,
     )
Example #8
0
 def plot_branch(
     self,
     colors: Optional[List[Optional[str]]] = None,
 ):
     """
     Plot multi-network ternary branch type plot.
     """
     branch_counts_list = [
         network.branch_counts for network in self.networks
     ]
     labels = [network.name for network in self.networks]
     return parameters.plot_ternary_plot(
         counts_list=branch_counts_list,
         labels=labels,
         is_nodes=False,
         colors=colors,
     )