def testUnpackOriginalDefaultsToIdentity(self): # Tensor. tensor = tf.fill([3, 4], 1.3) unpacked_tensor = input_utils.unpack_original_inputs(tensor) self.assertIs(tensor, unpacked_tensor) # Dict of Tensors. tensors = {"f1": tf.fill([1, 2], 2), "f2": tf.fill([4, 5], "a")} unpacked_tensors = input_utils.unpack_original_inputs(tensors) self.assertIs(tensors, unpacked_tensors) # Arbitrary object. obj = set(["a", "b", "c"]) unpacked_obj = input_utils.unpack_original_inputs(obj) self.assertIs(obj, unpacked_obj)
def testUnpackOriginal(self): # Tensor. tensor = tf.fill([3, 4], 1.3) packed_inputs = input_utils.MinDiffPackedInputs(tensor, None) unpacked_tensor = input_utils.unpack_original_inputs(packed_inputs) self.assertIs(unpacked_tensor, tensor) # Dict of Tensors. tensors = {"f1": tf.fill([1, 2], 2), "f2": tf.fill([4, 5], "a")} packed_inputs = input_utils.MinDiffPackedInputs(tensors, None) unpacked_tensors = input_utils.unpack_original_inputs(packed_inputs) self.assertIs(unpacked_tensors, tensors) # Arbitrary object. obj = set(["a", "b", "c"]) packed_inputs = input_utils.MinDiffPackedInputs(obj, None) unpacked_obj = input_utils.unpack_original_inputs(packed_inputs) self.assertIs(unpacked_obj, obj) # None. packed_inputs = input_utils.MinDiffPackedInputs(None, None) unpacked_obj = input_utils.unpack_original_inputs(packed_inputs) self.assertIsNone(unpacked_obj)