def test_save_input_values(self): inputs = [np.array([1, 2], dtype=np.int32)] self.assertEqual('2xi32=1 2', tf_utils.save_input_values(inputs)) inputs = [np.array([1, 2], dtype=np.float32)] self.assertEqual('2xf32=1.0 2.0', tf_utils.save_input_values(inputs))
def test_save_input_values(self, inputs, inputs_str): self.assertEqual(tf_utils.save_input_values(inputs), inputs_str)