← Back
Editing: custom_doctests.cpython-311.pyc
� �@�c � � � d Z d� Zd� ZdeiZdS )a Handlers for IPythonDirective's @doctest pseudo-decorator. The Sphinx extension that provides support for embedded IPython code provides a pseudo-decorator @doctest, which treats the input/output block as a doctest, raising a RuntimeError during doc generation if the actual output (after running the input) does not match the expected output. An example usage is: .. code-block:: rst .. ipython:: In [1]: x = 1 @doctest In [2]: x + 2 Out[3]: 3 One can also provide arguments to the decorator. The first argument should be the name of a custom handler. The specification of any other arguments is determined by the handler. For example, .. code-block:: rst .. ipython:: @doctest float In [154]: 0.1 + 0.2 Out[154]: 0.3 allows the actual output ``0.30000000000000004`` to match the expected output due to a comparison with `np.allclose`. This module contains handlers for the @doctest pseudo-decorator. Handlers should have the following function signature:: handler(sphinx_shell, args, input_lines, found, submitted) where `sphinx_shell` is the embedded Sphinx shell, `args` contains the list of arguments that follow: '@doctest handler_name', `input_lines` contains a list of the lines relevant to the current doctest, `found` is a string containing the output from the IPython shell, and `submitted` is a string containing the expected output from the IPython shell. Handlers must be registered in the `doctests` dict at the end of this module. c � � ddl }ddl m}m} | � d� � r | dd� } | � d� � r*|� t | � � t �� � }n"|� t | � � � � }|S ) aI Simplistic converter of strings from repr to float NumPy arrays. If the repr representation has ellipsis in it, then this will fail. Parameters ---------- s : str The repr version of a NumPy array. Examples -------- >>> s = "array([ 0.3, inf, nan])" >>> a = str_to_array(s) � N)�inf�nan�array� ����[)�dtype)�numpyr r � startswithr �eval�float� atleast_1d)�s�npr r �as �C/usr/lib/python3/dist-packages/IPython/sphinxext/custom_doctests.py� str_to_arrayr 3 s� � �" ���� � � � � � � � ��|�|�H� � � � �a��d�G���|�|�D��� $� �H�H�T�!�W�W�E�H� *� *��� �M�M�%��(�(� #� #�� �H� c �t �� ddl }t |� � dk rd}d}nc t |d � � }t |d � � }n7# t $ r*}d� |� � }t |� � |�d}~ww xY w t |� � }t |� � }|� |� � } |� |� � } |� | | � � } | |� || || ||�� � z } n # d } Y nxY wd �| j } | �d } d }n7| j j j } d � �fd �| j D � � � � }| rYd}|� | |d � |� � t |� � t |� � ��� � }t |� � �dS )z� Doctest which allow the submitted output to vary slightly from the input. Here is how it might appear in an rst file: .. code-block:: rst .. ipython:: @doctest float In [1]: 0.1 + 0.2 Out[1]: 0.3 r N� g�h㈵��>g:�0�yE>� zEBoth `rtol` and `atol` must be specified if either are specified: {0})�rtol�atolTz � Unavailable� c � �� g | ]}�|z ��S � r )�.0�line�TABs �r � <listcomp>z!float_doctest.<locals>.<listcomp>� s �� � F� F� F�D�S�4�Z� F� F� Fr z�doctest float comparison failure Document source: {0} Raw content: {1} On input line(s): {TAB}{2} we found output: {TAB}{3} instead of the expected: {TAB}{4} )r! )r �lenr � IndexError�formatr �isnan�allclose� directive�state�document�current_source�join�content�repr� RuntimeError)� sphinx_shell�args� input_lines�found� submittedr r r �e� found_isnan�submitted_isnan�errorr( �sourcer- r! s @r � float_doctestr: U s� �� � ���� �4�y�y�A�~�~������ '���a��>�>�D���a��>�>�D�D��� '� '� '�0�06��t� � � � �Q�-�-�Q� &����� '���� 7� ��+�+� � �U�#�#�� �h�h�u�o�o� � �(�(�9�-�-���K�K� �_�=�=�=�� �R�[�[�� �|�!4�!*�O�+;�!<�&*�� !� 7� 7� 7� 7������������ �C� �&�I��� �� ���� �)�8���)�)� F� F� F� F�I�4E� F� F� F�G�G�� � �6�� �H�H�V�W�d�i�i� �&<�&<�d�5�k�k��)�_�_�#� � /� /���1�o�o� � � s# �*A � A>�%A9�9A>� D � Dr N)�__doc__r r: �doctestsr r r �<module>r= sE ��0� 0�d � � �D@ � @ � @ �J �]� ���r
Save File
Cancel