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>OZread_stringnl_noescapeZpybytesZpylistZ read_bytes1Zread_bytearray8Z read_stringnlZstring1Zpybytes_or_strZ read_bytes4Zstring4Z pybytearrayZread_unicodestringnlZread_decimalnl_shortZpyboolZstringnl_noescapeZ markobjectZpysetZpybuffer bytes_typesZTAKEN_FROM_ARGUMENT1Zuint4Zunicodestring4Zbytes1Z read_string1Zunicodestring1ZfloatnlZpyfloatZ anyobjectZpytupleZread_decimalnl_longZuint8Z UP_TO_NEWLINEZ read_bytes8Z read_uint4ZunicodestringnlZbytes4Z read_long4Zunicodestring8ZopcodesZ bytearray8Z stacksliceZdecimalnl_shortZ StackObjectZTAKEN_FROM_ARGUMENT4Z read_floatnlZfloat8Z read_uint2ZArgumentDescriptorZbytes8Zlong1Z OpcodeInfoZ read_string4Zuint2Zuint1Zread_unicodestring8ZstringnlZ read_uint1ZTAKEN_FROM_ARGUMENT8UZ read_uint8Zint4ZpystringZpyintZlong4ZTAKEN_FROM_ARGUMENT4UZpynoneZpydictZread_unicodestring1Zstringnl_noescape_pairZ read_long1ZpylongZ read_float8Zcode2opZread_unicodestring4Z pyunicodeZ pyfrozensetZpyinteger_or_boolZread_stringnl_noescape_pairZdecimalnl_longZ read_int4) 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