o
    
i
                     @   sr   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
abcxyzc                  C   s   ddl m}  |  S )Nr   YAML)srsly.ruamel_yamlr   r
    r   _/home/ubuntu/.local/lib/python3.10/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  t}t| tt|d  |tksJ d S )Nr   )r   load
single_docprinttypesingle_data)selfdr   r   r   test_single_load+   s   zTestOldStyle.test_single_loadc                 C   sB   t t t  tksJ W d    d S 1 sw   Y  d S N)pytestraises	TypeErrorr   r   r   )r   r   r   r   test_single_load_no_arg1   s   "z$TestOldStyle.test_single_load_no_argc                 C   s    t t t}|tksJ d S r   )listr   load_all	multi_docmulti_doc_data)r   datar   r   r   test_multi_load5   s   zTestOldStyle.test_multi_loadc                 C   s,   t  ttj | \}}|tksJ d S r   )r   dumpr   sysstdout
readouterrr   )r   capsysouterrr   r   r   test_single_dump9   s   zTestOldStyle.test_single_dumpc                 C   s6   t  }d|_|ttj | \}}|tksJ d S )NT)r   explicit_startdump_allr!   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   sf   ddl m} |tjd}|t W d    n1 sw   Y  | \}}t| |tks1J d S )Nr   r
   output)	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   s~   ddl m} |tjd}d|_|td  |td  W d    n1 s(w   Y  | \}}t| |t	ks=J d S )Nr   r
   r4   Tr   )
r   r   r%   r&   r,   r$   r!   r'   r   r    r6   r   r   r   r/   P   s   z"TestContextManager.test_multi_dumpc                 C   s|   ddl m} |tjd}d|_|tD ]}|| qW d    n1 s'w   Y  | \}}t	| |tks<J d S )Nr   r
   r4   T)
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)r0   r1   r2   r+   r/   r7   r   r   r   r   r3   F   s    	r3   )
__future__r   r%   r   r   dictr   r    r!   r   r   r3   r   r   r   r   <module>   s   