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)
