def testUnpackMinDiffData(self): # Tensor. tensor = tf.fill([3, 4], 1.3) packed_inputs = input_utils.MinDiffPackedInputs(None, tensor) unpacked_tensor = input_utils.unpack_min_diff_data(packed_inputs) self.assertIs(unpacked_tensor, tensor) # Tuple of Tensors. tensors = ({ "f1": tf.fill([1, 2], 2), "f2": tf.fill([4, 5], "a") }, None, tf.fill([4, 1], 1.0)) packed_inputs = input_utils.MinDiffPackedInputs(None, tensors) unpacked_tensors = input_utils.unpack_min_diff_data(packed_inputs) self.assertIs(unpacked_tensors, tensors) # Arbitrary object. obj = set(["a", "b", "c"]) packed_inputs = input_utils.MinDiffPackedInputs(None, obj) unpacked_obj = input_utils.unpack_min_diff_data(packed_inputs) self.assertIs(unpacked_obj, obj) # None. packed_inputs = input_utils.MinDiffPackedInputs(None, None) unpacked_obj = input_utils.unpack_min_diff_data(packed_inputs) self.assertIsNone(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)