
    VhO                     6    d dl Zd dlmZ d Zd Zd Zd Zd Zy)    N)junction_treec                      t        j                         } | j                  g d       t        |       }t        j                         }|j                  g d       t        j
                  ||      sJ y )N)ACBr   r   Dr   E))r   r   )r   r   r	   r   )r   r   r   r
   nxDiGraphadd_edges_fromr   Graphis_isomorphicr	   GJs      a/home/dcms/DCMS/lib/python3.12/site-packages/networkx/algorithms/tree/tests/test_decomposition.py'test_junction_tree_directed_confoundersr      s\    


AEFaA

A	
 Aq!!!    c                      t        j                         } | j                  dg       t        |       }t        j                         }|j                  dg       t        j
                  ||      sJ y )N)r   r	   r   r   )r   r   add_nodes_fromr   r   r   r   s      r   -test_junction_tree_directed_unconnected_nodesr      s\    


A*+,aA

A*+,Aq!!!r   c                      t        j                         } | j                  g d       t        |       }t        j                         }|j                  g d       t        j
                  ||      sJ y )N)r   r	   r   r
   ))r!   r	   )r"   r   r   r   r   r   r   s      r   #test_junction_tree_directed_cascader$   "   s\    


A9:aA

A	
 Aq!!!r   c                      t        j                         } | j                  g d       t        |       }t        j                         }|j                  g d       t        j                  ||      sJ y )N)r!   r
   )r   F)r   r   r   r   r   r   r   r   s      r   -test_junction_tree_directed_unconnected_edgesr'   3   sV    


A9:aA

A9:Aq!!!r   c                      t        j                         } | j                  g d       t        |       }t        j                         }|j                  g d       t        j                  ||      sJ y )N)r   r   r   r   r   ))r)   r   )r*   r   )r   r   )r   r   r#   )r   r   )r   r   r   r   r   r   s      r   test_junction_tree_undirectedr+   >   s\    

AEFaA

A	
	 Aq!!!r   )	networkxr   &networkx.algorithms.tree.decompositionr   r   r   r$   r'   r+    r   r   <module>r/      s$     @"$"""""r   