Format: application/x-python-code
1 o
2
3 ��h�- � @ sT d Z ddlmZ ddlmZ ddlZe�d�Ze�d� ddl m
4 Z
5 ddlm Z dd l
6 mZ ddlZdd
7 lmZ dd lmZ dd lmZmZ d
8 dlmZ ddlm Z dd� Z dd� Z e � d�dd� �Z!dd� Z"ddddd d �Z#d d � Z$d d!� Z%d"d#� Z&d$d%� Z'd&d'� Z(G d(d)� d)e)�Z*d*d+� Z+d,d-� Z,ej-�.d.g d/��d0d1� �Z/d2d3� Z0d4d5� Z1dS )6z Tests for pylab tools module.
9 � )�
10 a2b_base64)�BytesION�
11 matplotlibZAgg)�Figure)�pyplot)�backend_inline)� get_ipython)�InteractiveShell)�_PNG�_JPEG� )�
12 pylabtools)�
13 decoratorsc C sz t �� } t�| d�d u sJ �t �d� t �� } | �ddd�}|�g d�� t �� t�| d�d d� �� }d|v s;J �d S )N�svg�all� �r r � �d z doctype svg) �plt�figure�pt� print_figure�close� add_subplot�plot�draw�lower)�fig�axr � r �Xc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/core/tests/test_pylabtools.py�test_figure_to_svg s
14 r" c
15 C sf ddl m} t� }| �dd�}z |�|d� W dS ty2 } z|jj}t� d||f �|�d}~ww )z0Skip if PIL can't write JPEGs to BytesIO objectsr )�Image�RGB)� r% �jpegz'PIL can't write JPEG to BytesIO: %s: %sN)
16 ZPILr# r �new�save� Exception� __class__�__name__�pytest�skip)r# �bufZimg�eZenamer r r! �_check_pil_jpeg_bytes- s ��r0 z PIL.Imagec C sd t � t�� } | �ddd�}|�g d�� t�� tj| dddid�d d� �� }|� t
17 �s0J �d S )Nr r r& �optimize�2 )Z
18 pil_kwargsr ) r0 r r r r r r r r �
19 startswithr )r r r& r r r! �test_figure_to_jpeg: s r4 c C s� t �� } t�| �d ks
20 J �t �d� t �� } | �ddd�}|�g d�� t �� t�| �\}}|�t �s6J �d|v s<J �d|v sBJ �d S )Nr r r �width�height)
21 r r r Z
22 retina_figurer r r r r3 r
23 )r r �png�mdr r r! �test_retina_figureE s
24 r9 � image/pngz
25 image/jpegzapplication/pdfz
26 image/svg+xml)r7 r& �pdf�retinar c C s` t � } t�� D ]&\}}t�| |� | jj�� D ]\}}||kr&t|v s%J �qt|vs,J �qqd S �N)r �
27 _fmt_mime_map�itemsr �select_figure_formats�display_formatter�
28 formattersr )�ip�fmtZ active_mime�mime�fr r r! � test_select_figure_formats_str] s ��rG c
29 C s� t � } tdd�}tj| dfi |�� | jjd }|�t�}|j}|}d|d< d|d< ||ks0J �t �
30 � }|� ddd�}|� g d �� t �
31 � d|_||�}t|t�sTJ �t|�} | �t�s_J �d S )
32 NZtight)Z bbox_inchesr7 r: T�base64rD r r )r �dictr r@ rA rB Zlookup_by_typer �keywordsr r r r r �enabled�
33 isinstance�strr r3 r
34 )
35 rC �kwargs� formatterrF �cell�expectedr r r7 Z png_bytesr r r! �!test_select_figure_formats_kwargsg s&
36
37 rR c C sv t � } ddhdgddhfD ]+}dd� |D �}t�| |� | jj�� D ]\}}||v r1t|v s0J �q"t|vs7J �q"q
38 d S )Nr7 r )r& r; r< c S � h | ]}t | �qS r �r> ��.0rD r r r! � <setcomp>� � z1test_select_figure_formats_set.<locals>.<setcomp>)r r r@ rA rB r? r )rC �fmts� active_mimesrE rF r r r! � test_select_figure_formats_set~ s � ��r[ c C s� t � } t�t�� t�| d� W d � n1 sw Y t�t�� t�| ddh� W d � n1 s6w Y t�t�� t�| g d�� W d � d S 1 sTw Y d S )NZfoor7 )r< r; �bar�bad)r r, �raises�
39 ValueErrorr r@ )rC r r r! � test_select_figure_formats_bad� s � � "�r` c C s2 i } t j| dd� d| v sJ �| d tksJ �d S )NF)Z
40 import_allr �np)r � import_pylabra )�nsr r r! �test_import_pylab� s rd c @ sd e Zd ZG dd� de�Zdd� Zdd� Zdd� Zd d
41 � Zd d � Z d
42 d� Z
43 e � e
44 �� �dd� �ZdS )�TestPylabSwitchc s$ e Zd Z� fdd�Zdd� Z� ZS )zTestPylabSwitch.Shellc s d| j j_t� �� dS )z:Sets up the command history, and starts regular autosaves.z:memory:N)�configZHistoryManagerZ hist_file�super� init_history)�self�r* r r! rh � s
45 z"TestPylabSwitch.Shell.init_historyc C � d S r= r )ri �guir r r! �
46 enable_gui� s z TestPylabSwitch.Shell.enable_gui)r+ �
47 __module__� __qualname__rh rm �
48 __classcell__r r rj r! �Shell� s rq c sr dd l � � fdd�}� j| _� j| _tdd�� _tdd�� _tj| _|t_tj | _
49 dd� t_ t j | _
50 dd� t _ d S ) Nr c s | � j d< d S )N�backend)�rcParams�rr �r r r! �act_mpl� s z-TestPylabSwitch.setup_method.<locals>.act_mplZQtAggrt c _ rk r= r ��a�kwr r r! �<lambda>� � z.TestPylabSwitch.setup_method.<locals>.<lambda>c _ rk r= r rw r r r! rz � r{ )r rs �_saved_rcParams� rcParamsOrig�_saved_rcParamsOrigrI r �activate_matplotlib�_save_amrb �_save_ipr �configure_inline_support� _save_cis)ri rv r ru r! � setup_method� s
51 z TestPylabSwitch.setup_methodc C s4 | j t_| jt_| jt_dd l}| j |_
52 | j |_ d S )Nr )
53 r� r r r� rb r� r r� r r| rs r~ r} )ri r r r r! �teardown_method� s z TestPylabSwitch.teardown_methodc C s� | � � }|�d �\}}|dksJ �|jdksJ �|�d�\}}|d u s%J �|jdks,J �|�d�\}}|dks9J �|jdks@J �|�d�\}}|d u sMJ �|jdksTJ �|�� \}}|dks`J �|jdksgJ �d S )N�qt�inline�rq �enable_matplotlibZpylab_gui_select�ri �srl rr r r r! �test_qt� s zTestPylabSwitch.test_qtc C s� | � � }|�d�\}}|d u sJ �|jd ksJ �|�d�\}}|d u s%J �|jd ks,J �|�d�\}}|dks9J �|jdks@J �d S )Nr� r� r� r� r r r! � test_inline� s zTestPylabSwitch.test_inlinec C s� | � � }|�d�\}}|du sJ �dh}dd� |D �}t�||� |�d�\}}|du s.J �|jj�� D ]\}}||v rCt|v sBJ �q4t|vsIJ �q4dS )z<Using '%matplotlib inline' twice should not reset formattersr� Nr7 c S rS r rT rU r r r! rW � rX z4TestPylabSwitch.test_inline_twice.<locals>.<setcomp>)rq r� r r@ rA rB r? r )ri rC rl rr rY rZ rE rF r r r! �test_inline_twice� s �z!TestPylabSwitch.test_inline_twicec C s\ | � � }|�d�\}}|dksJ �|jdksJ �|�d�\}}|dks%J �|jdks,J �d S )Nr� �gtk3r� r� r r r! � test_qt_gtk� s zTestPylabSwitch.test_qt_gtkc C s� d}d}t jdk}| �� }|�|� |r/t�t��
54 |�|� W d � n1 s)w Y n|�|� |�d|� |rZt�t�� |�d|� W d � d S 1 sSw Y d S |�d|� d S )N�)module://matplotlib_inline.backend_inlinez)module://matplotlib_inline.Backend_inline)r � r r )r �__version_info__rq r� r, r^ � RuntimeErrorZrun_line_magic)ri Z
55 all_lowercaseZsome_uppercaseZmpl3_9_1r� r r r! �'test_backend_module_name_case_sensitive s"
56
57 ��
58 "�z7TestPylabSwitch.test_backend_module_name_case_sensitiveN)r+ rn ro r rq r� r� r� r� r� r� �decZskipifr � _matplotlib_manages_backendsr� r r r r! re � s
59 re c C s dD ] } | t jvs J �qd S )N)�aggr r; �ps)r Z backend2gui)�kr r r! �test_no_gui_backends s �r� c C s t � } d | _t�| � d S r= )r Zcanvasr r )r r r r! �test_figure_no_canvas" s r� z$name, expected_gui, expected_backend) )r� r� �gtk3agg)�gtk4r� �gtk4agg)�headlessNr� )�osxr� �macosx)r� r� �qtagg)�qt5r� �qt5agg)�qt6r� r� )�tkr� �tkagg)�wxr� �wxagg)r� Nr� )�cairoNr� )r; Nr; )r� Nr� )r Nr )�templateNr� )r� r� r� )� gtk3cairor� r� )r� r� r� )� gtk4cairor� r� )r� r� r� )�nbaggr� r� )�notebookr� r� )r� r� r� )�qtcairor� r� )r� r� r� )�qt5cairor� r� )r� r� r� )�tkcairor� r� )�webaggr� r� )r� r� r� )�wxcairor� r� c C s� t �� }|s)| �d�s| �d�s| dv rt�� n| dkr d}n | dkr)||}}t �| �\}}|sB|r8|�� nd }|r@|�� nd }||ksHJ �||ksNJ �d S )Nr� r� )r� r� r; r� r r� r� r� r� )r r� �endswithr, r- �find_gui_and_backendr )�nameZ expected_gui�expected_backendZmpl_manages_backendsrl rr r r r! �test_backend_builtin( s$ +��
60
61 r� c C s: t �d�\} }| d u s
62 J �t �� rdnd}||ksJ �d S )Nr� r� )r r� r� )rl rr r� r r r! �test_backend_entry_pointj s ��r� c C sF t �t�� rtnt�� t�d� W d � d S 1 s w Y d S )Nzname-does-not-exist)r, r^ r r� r� �KeyErrorr� r r r r! �test_backend_unknownu s "�r� )2�__doc__�binasciir �ior r, Z importorskipr ZuseZmatplotlib.figurer r r Zmatplotlib_inliner �numpyra ZIPython.core.getipythonr Z IPython.core.interactiveshellr ZIPython.core.displayr
63 r � r
64 r ZIPython.testingr r� r" r0 Z skip_withoutr4 r9 r> rG rR r[ r` rd �objectre r� r� �markZ parametrizer� r� r� r r r r! �<module> sR
65
66
67
68 �
69 �
70 &