Example #1
0
 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))
Example #2
0
 def test_save_input_values(self, inputs, inputs_str):
     self.assertEqual(tf_utils.save_input_values(inputs), inputs_str)