o i e@sLdZddlZddlZddlmmZGdddZGdddej j Z dS)z%Tests utils for preprocessing layers.Nc@seZdZddZddZdS) ArrayLikecCs ||_dSN)values)selfrrf/var/www/myenv/lib/python3.10/site-packages/keras/src/layers/preprocessing/preprocessing_test_utils.py__init__s zArrayLike.__init__cCs t|jSr)nparrayr)rrrr __array__s zArrayLike.__array__N)__name__ __module__ __qualname__rr rrrrrs rc@sPeZdZdZdddZdddZeZddZd d Zd d Z d dZ ddZ dS)PreprocessingLayerTestz7Base test class for preprocessing layer API validation.NcCs |dus|dur|j|||ddSt|ttfr7|t|t|t||D] \}}|j|||dq(dSt|tj j rj|t|t|| D]\}}||}|r^|d|dnd}||||qLdSt|t s|t |drt|jtjr|j|||ddS|j|||ddS)zBAsserts that elements are close (if numeric) or equal (if string).Nmsgz ()dtype)assertAllEqual isinstancelisttuple assertEquallenzipassertAllCloseOrEqual collectionsabcMappingitemsfloathasattrr issubdtypernumberassertAllClose)rabra_valueb_valuekey error_messagerrrr%s,z,PreprocessingLayerTest.assertAllCloseOrEqualcCs(||}||}|j|||ddS)Nr)extractr)rcombineracc1acc2rdata_1data_2rrrassert_extracted_output_equal<s  z4PreprocessingLayerTest.assert_extracted_output_equalc Cst|dkrtdt|dt|dg}t|dg}t|dd}||}|||||||g}|j||dd |||||||g} |j|| d d ||||||||gg} |j|| d d ||||||} |j|| d d |||||||g} |j|| d d ||||g||||g} |j|| dd |j||dd dS)zNValidate that various combinations of compute and merge are identical.z7Data must have at least 4 elements. Received len(data)=.rNz0Sharding data should not change the data output.rz?The order of merge arguments should not change the data output.z:Nesting merge arguments should not change the data output.zrestore loop loses no data.N)r7r+restorer1r)rr,r:r;rErF restored_accrrr(validate_accumulator_extract_and_restores   z?PreprocessingLayerTest.validate_accumulator_extract_and_restorecCs:||}||}||}||||||dS)z=Validate that the serialize<->deserialize loop loses no data.N)r7 serialize deserializer9)rr,r:r;rEserialized_datadeserialized_datarrr.validate_accumulator_serialize_and_deserializes    zEPreprocessingLayerTest.validate_accumulator_serialize_and_deserializecCs0||}||}||||||dS)zAValidate that every call to compute creates a unique accumulator.N)r7 assertIsNotr9)rr,r:rEr.rrrvalidate_accumulator_uniquenesss   z6PreprocessingLayerTest.validate_accumulator_uniquenessr) r r r__doc__rr1r9rDrGrJrOrQrrrrrs  g r) rRrnumpyr tensorflow.compat.v2compatv2tfrtestTestCaserrrrrs