o
    3ήc                     @   s&   d Z ddlZdd ZG dd dZdS )z)Unit tests for the beam search functions.    Nc                 C   s   | S )N )xr   r   Z/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/traversal/tests/test_beamsearch.pyidentity   s   r   c                   @   s    e Zd ZdZdd Zdd ZdS )TestBeamSearchz(Unit tests for the beam search function.c                 C   s4   t d}t j|dtdd}t|ddgksJ dS )z>Tests that a narrow beam width may cause an incomplete search.   r      widthr      r      Nnxcycle_graphbfs_beam_edgesr   listselfGedgesr   r   r   test_narrow   s   
zTestBeamSearch.test_narrowc                 C   s4   t d}t j|dtdd}t|g dksJ d S )Nr   r   r   r	   )r   )r   r   r   r   r   r   r   r   	test_wide   s   
zTestBeamSearch.test_wideN)__name__
__module____qualname____doc__r   r   r   r   r   r   r   
   s    r   )r   networkxr   r   r   r   r   r   r   <module>   s    