3
d
                 @   sp   d dl mZ d dlZd dlZdZeddeddgdgZd	Zd
dgegZdd Z	G dd dZ
G dd dZdS )    )print_functionNz- a: 1
- b:
  - 2
  - 3
   )a      )bz,---
- abc
- xyz
---
- a: 1
- b:
  - 2
  - 3
abcZxyzc              C   s   ddl m}  |  S )Nr   )YAML)srsly.ruamel_yamlr	   )r	    r   _/var/www/html/virt/lib64/python3.6/site-packages/srsly/tests/ruamel_yaml/test_contextmanager.pyget_yaml$   s    r   c               @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestOldStylec             C   s4   t  jt}t| tt|d  |tks0td S )Nr   )r   load
single_docprinttypesingle_dataAssertionError)selfdr   r   r   test_single_load+   s    zTestOldStyle.test_single_loadc          	   C   s,   t jt t j tkstW d Q R X d S )N)pytestZraises	TypeErrorr   r   r   r   )r   r   r   r   test_single_load_no_arg1   s    z$TestOldStyle.test_single_load_no_argc             C   s    t t jt}|tkstd S )N)listr   load_all	multi_docmulti_doc_datar   )r   datar   r   r   test_multi_load5   s    zTestOldStyle.test_multi_loadc             C   s,   t  jttj |j \}}|tks(td S )N)r   dumpr   sysstdout
readouterrr   r   )r   capsysouterrr   r   r   test_single_dump9   s    zTestOldStyle.test_single_dumpc             C   s6   t  }d|_|jttj |j \}}|tks2td S )NT)	r   explicit_startZdump_allr   r"   r#   r$   r   r   )r   r%   yamlr&   r'   r   r   r   test_multi_dump>   s
    zTestOldStyle.test_multi_dumpN)__name__
__module____qualname__r   r   r    r(   r+   r   r   r   r   r   *   s
   r   c               @   s$   e Zd Zdd Zdd Zdd ZdS )TestContextManagerc          	   C   sR   ddl m} |tjd}|jt W d Q R X |j \}}t| |tksNt	d S )Nr   )r	   )output)
r
   r	   r"   r#   r!   r   r$   r   r   r   )r   r%   r	   r*   r&   r'   r   r   r   r(   G   s    z#TestContextManager.test_single_dumpc          
   C   sj   ddl m} |tjd(}d|_|jtd  |jtd  W d Q R X |j \}}t| |t	ksft
d S )Nr   )r	   )r0   Tr   )r
   r	   r"   r#   r)   r!   r   r$   r   r   r   )r   r%   r	   r*   r&   r'   r   r   r   r+   P   s    z"TestContextManager.test_multi_dumpc          
   C   sl   ddl m} |tjd*}d|_x|jtD ]}|j| q,W W d Q R X |j \}}t	| |tksht
d S )Nr   )r	   )r0   T)r
   r	   r"   r#   r)   r   r   r!   r$   r   r   )r   r%   r	   r*   r   r&   r'   r   r   r   test_roundtripl   s    z!TestContextManager.test_roundtripN)r,   r-   r.   r(   r+   r1   r   r   r   r   r/   F   s   	r/   )
__future__r   r"   r   r   dictr   r   r   r   r   r/   r   r   r   r   <module>   s   