ó
šxŠ\c        	   @   sb  d  d l  Z  d  d l Z d  d l Z d  d l Z d  d l Z d  d l j j Z	 d  d l
 j j Z e  j d „  ƒ Z e j j d e j d d d d g d d	 d
 g g d d d d g ƒe j d d d d g d d ƒg ƒ e j j d d d d g ƒ d „  ƒ ƒ Z e j j d e j d d d d g d d	 d
 g g d d d d g ƒe j d d d d g d d ƒg ƒ e j j d d d g ƒ d „  ƒ ƒ Z e j j d d i e d 6e j f d i  e j f d i  e j f g ƒ d „  ƒ Z e j j d d i e d 6e d 6e j i e d 6f d i  e j i d d 6f d i  e j i  f g ƒ d „  ƒ Z d „  Z d S(    iÿÿÿÿNc           c   s2   t  j d t ƒ  t  j d t ƒ d  VWd  QXd  S(   Nt   recordt   ignore(   t   warningst   catch_warningst   Truet   simplefiltert   FutureWarning(    (    (    s?   lib/python2.7/site-packages/pandas/tests/io/test_compression.pyt   catch_to_csv_depr   s    t   objid   g¿¶~úÏš¿?g9DÜœJÎ?g>éD‚)â?gÜ.4×i¤(@g33333þ@gÍÌÌÌ¤œAt   columnst   Xt   Yt   Zt   namet   methodt	   to_picklet   to_jsont   to_csvc         C   s‘   t  j ƒ   } t ƒ  m t |  | ƒ | d | ƒt j j | ƒ } t |  | ƒ | d d  ƒt j j | ƒ } | | k s t ‚ Wd  QXWd  QXd  S(   Nt   compression(	   t   tmt   ensure_cleanR   t   getattrt   ost   patht   getsizet   Nonet   AssertionError(   R   R   t   compression_onlyR   t   compressed_sizet   uncompressed_size(    (    s?   lib/python2.7/site-packages/pandas/tests/io/test_compression.pyt   test_compression_size   s    
c         C   s>  t  j ƒ  † } t j | d d | ƒ\ } } t ƒ  V | ( t |  | ƒ | ƒ | j sa t ‚ Wd  QX| j sv t ‚ t j	 j
 | ƒ } Wd  QXWd  QXt  j ƒ  ˜ } t j | d d d  ƒ\ } } t ƒ  5 | ( t |  | ƒ | ƒ | j sõ t ‚ Wd  QXWd  QX| j st ‚ t j	 j
 | ƒ } | | k s4t ‚ Wd  QXd  S(   Nt   wR   (   R   R   t   icomt   _get_handleR   R   t   closedR   R   R   R   R   (   R   R   R   R   t   ft   handlesR   R   (    (    s?   lib/python2.7/site-packages/pandas/tests/io/test_compression.pyt   test_compression_size_fh(   s"    

s'   write_method, write_kwargs, read_methodt   indexc         C   s›   t  j d d d g d d d g g d d d	 d
 g ƒ} t j | } t j d | ƒ / } t | |  ƒ | |  | | d | ƒ} Wd  QXt j | | ƒ d  S(   Ng      ð?i    iüÿÿÿg333333@i   i   R	   R
   R   R   t
   compressedR   (   t   pdt	   DataFrameR    t   _compression_to_extensionR   R   R   t   assert_frame_equal(   t   write_methodt   write_kwargst   read_methodR   t   inputt	   extensionR   t   output(    (    s?   lib/python2.7/site-packages/pandas/tests/io/test_compression.pyt,   test_dataframe_compression_defaults_to_inferB   s    6s1   write_method,write_kwargs,read_method,read_kwargst   headert   squeezet   seriest   typc   	   	   C   s   t  j d d d d g d d ƒ} t j | } t j d | ƒ 2 } t | |  ƒ | |  | | d | | } Wd  QXt j | | d	 t ƒd  S(
   Ni    i   iþÿÿÿi
   R   R
   R'   R   t   check_names(	   R(   t   SeriesR    R*   R   R   R   t   assert_series_equalt   False(	   R,   R-   R.   t   read_kwargsR   R/   R0   R   R1   (    (    s?   lib/python2.7/site-packages/pandas/tests/io/test_compression.pyt)   test_series_compression_defaults_to_inferR   s    
!c         C   s­   t  j d d d d g d d d g g d d	 d
 d g ƒ} t j ƒ  a } t j | d d |  ƒ\ } } t j t d t ƒ% |  | j	 | d |  ƒWd  QXWd  QXWd  QXd  S(   Nid   g¿¶~úÏš¿?g9DÜœJÎ?g>éD‚)â?gÜ.4×i¤(@g33333þ@gÍÌÌÌ¤œAR	   R
   R   R   R   R   t   check_stacklevel(
   R(   R)   R   R   R    R!   t   assert_produces_warningt   RuntimeWarningR:   R   (   R   t   dfR   R#   R$   (    (    s?   lib/python2.7/site-packages/pandas/tests/io/test_compression.pyt   test_compression_warningd   s    
(   t
   contextlibR   R   t   pytestt   pandasR(   t   pandas.util.testingt   utilt   testingR   t   pandas.io.commont   iot   commonR    t   contextmanagerR   t   markt   parametrizeR)   R8   R   R%   R:   t   read_csvt	   read_jsont   read_pickleR2   R   R<   RA   (    (    (    s?   lib/python2.7/site-packages/pandas/tests/io/test_compression.pyt   <module>   s:   %*%'!$