Your IP : 216.73.216.85
���]| � @ s� d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z Gd d � d e e � Z d d � Z e
d k r� e � d S)
� N)�support)�AbstractPickleTests)�AbstractPickleModuleTestsc @ sI e Z d Z d d d � Z d d � Z d Z d d � Z d d � Z d S)
�OptimizedPickleTestsNc C s t j t j | | � � S)N)� pickletools�optimize�pickle�dumps)�self�arg�proto� r
�5/opt/alt/python35/lib64/python3.5/test_pickletools.pyr
s zOptimizedPickleTests.dumpsc K s t j | | � S)N)r �loads)r
�buf�kwdsr
r
r r
s zOptimizedPickleTests.loadsc C s d d � t d � D� } | j | d � x� t t j d � D]� } t j | | � } t j | � } | j | | � | j | d | d � t j | � } t j | � } | j | | � | j | d | d
� | j
t j | � | j
t j | � q> Wd S) Nc S s g | ] } t | � � q Sr
)�str)�.0�ir
r
r �
<listcomp> s zBOptimizedPickleTests.test_optimize_long_binget.<locals>.<listcomp>i � � ���r ���r r )
�range�appendr �HIGHEST_PROTOCOLr r � assertEqual�assertIsr r � assertNotIn� LONG_BINGET� LONG_BINPUT)r
�datar �pickled� unpickled�pickled2�
unpickled2r
r
r �test_optimize_long_binget s z.OptimizedPickleTests.test_optimize_long_bingetc C s� d } | j t j | � t j | � } | j | d d d g � | j | d | d � t j | � } t j | � } | j | d d d g � | j | d | d � | j t j | � d S)Ns �� ]�(�spamq�ham�he.ZspamZhamr r ) ZassertInr �BINPUTr r r r r r )r
r# r$ r% r& r
r
r � test_optimize_binput_and_memoize# s z5OptimizedPickleTests.test_optimize_binput_and_memoize)�__name__�
__module__� __qualname__r r Ztest_pickle_to_2xr' r) r
r
r
r r s
r c C s t j t � t j t � d S)N)r Z run_unittestr Z run_doctestr r
r
r
r � test_main> s
r- �__main__) �structr r Ztestr Ztest.pickletesterr r r r- r* r
r
r
r �<module> s 6