a 5dgR@slddlZddlZddlmZddlmZddlZGdddeZGdddejZ dd Z e d krhe dS) N)support)AbstractPickleTestsc@s6eZdZd ddZddZdZdZddZdd ZdS) OptimizedPickleTestsNcKsttj||fi|SN) pickletoolsoptimizepickledumps)selfargprotokwargsr1/usr/local/lib/python3.9/test/test_pickletools.pyr szOptimizedPickleTests.dumpscKstj|fi|Sr)rloads)r bufkwdsrrrr szOptimizedPickleTests.loadscCsddtdD}||dttjdD]}t||}t|}|||||d|dt |}t|}|||||d|d| tj || tj |q.dS)NcSsg|] }t|qSr)str).0irrr zBOptimizedPickleTests.test_optimize_long_binget..i) rangeappendrHIGHEST_PROTOCOLr r assertEqualassertIsrr assertNotIn LONG_BINGET LONG_BINPUT)r datar pickled unpickledpickled2 unpickled2rrrtest_optimize_long_bingets      z.OptimizedPickleTests.test_optimize_long_bingetcCsd}|tj|t|}||gd||d|dt|}t|}||gd||d|d|tj|dS)Ns ](spamqhamhe.)Zspamhamr)r) ZassertInrBINPUTrrrrrr )r r$r%r&r'rrr test_optimize_binput_and_memoize%s   z5OptimizedPickleTests.test_optimize_binput_and_memoize)N) __name__ __module__ __qualname__r rZtest_pickle_to_2xZ+test_framed_write_sizes_with_delayed_writerr(r,rrrrrs  rc@seZdZddZdS) MiscTestCasecCshd}tj|t|ddS)N>OZpylongZ read_string1Zstring4ZpybytesZ read_long4Zread_decimalnl_shortZopcodesZstringnl_noescape_pairZlong4Z StackObjectZpyintZlong1ZpystringZ read_bytes8Z markobjectZ UP_TO_NEWLINEZread_unicodestring4Zdecimalnl_longZunicodestringnlZ stacksliceZuint8Zunicodestring4Z bytearray8 bytes_typesZcode2opZunicodestring8ZTAKEN_FROM_ARGUMENT8UZstringnl_noescapeZ pyunicodeZ read_string4Z read_float8Zread_stringnl_noescapeZread_stringnl_noescape_pairZpynoneZbytes8Zread_decimalnl_longZ read_floatnlZread_unicodestring1Z read_bytes1ZpylistZ pybytearrayZstringnlZ read_int4Zread_unicodestring8ZfloatnlZbytes1Zread_unicodestringnlZpytupleZ read_stringnlZpybufferZfloat8ZpyboolZ anyobjectZ read_bytes4Z read_uint4Zunicodestring1ZTAKEN_FROM_ARGUMENT1Zstring1Zuint4Z read_uint1Z pyfrozensetZ read_uint2Z read_long1ZpysetZuint1Zuint2ZTAKEN_FROM_ARGUMENT4Zpybytes_or_strZint4Zdecimalnl_shortZpyinteger_or_boolZread_bytearray8ZpyfloatZbytes4Z OpcodeInfoZ read_uint8ZArgumentDescriptorZTAKEN_FROM_ARGUMENT4UZpydict) blacklist)rZ check__all__r)r r2rrr test__all__AszMiscTestCase.test__all__N)r-r.r/r3rrrrr0@sr0cCs"ttttttdSr)rZ run_unittestrr0Z run_doctestrrrrr test_main`s  r4__main__) rrtestrZtest.pickletesterrZunittestrZTestCaser0r4r-rrrrs  9