a ߙfb@sJddlZddlmZmZddZGdddZddZd d Zd d ZdS) N)fnpickle fnunpicklecCst|d}d}t||t|d}||ks2Jt|d}t||Wdn1s\0Yt|d$}t|}||ksJWdn1s0YdS)z Tests the `fnpickle` and `fnupickle` functions' basic operation by pickling and unpickling a string, using both a filename and a file. z test1.pickleastringrwbNrb)strjoinrropen)tmpdirfnobj1resfrHlib/python3.9/site-packages/astropy/io/misc/tests/test_pickle_helpers.pytest_fnpickling_simples    ( rc@seZdZddZddZdS) ToBePickledcCs ||_dS)N)item)selfrrrr__init__szToBePickled.__init__cCst|tr|j|jkSdSdS)NF) isinstancerr)rotherrrr__eq__!s  zToBePickled.__eq__N)__name__ __module__ __qualname__rrrrrrrsrcCs<t|d}d}t|}t||t|}||ks8JdS)zl Tests the `fnpickle` and `fnupickle` functions' ability to pickle and unpickle custom classes. z test2.picklerN)rrrrr)r r r obj2r rrrtest_fnpickling_class(s  rcCsdddl}d}t|}t|jdD]<}t|d|d}t|||dt|}||ks"Jq"dS)z| Tests the `fnpickle` and `fnupickle` functions' ability to pickle and unpickle pickle files from all protcols. rNrZtestpz.pickle)Zprotocol)picklerrangeZHIGHEST_PROTOCOLrrrr)r rr rpr r rrrtest_fnpickling_protocol6sr"cCst|d}d}d}t||t||ddt|dd}t|dksJJ|d |ksZJ|d |ksjJt||ddt|dd}t|dksJttt|d dWd n1s0Yd S) z Tests the `fnpickle` and `fnupickle` functions' ability to pickle and unpickle multiple objects from a single file. z test3.picklegH}t@Z blahblahfooT)append)ZnumberrrN)rrrrlenpytestZraisesEOFError)r r Zobj3Zobj4r rrrtest_fnpickling_manyGs    r*) r(Zastropy.io.miscrrrrrr"r*rrrrs