Ejemplo n.º 1
0
 def testInputPreProcessErrorBadFormat(self):
   input_str = 'inputx=file[[v1]v2'
   with self.assertRaises(RuntimeError):
     saved_model_cli.preprocess_inputs_arg_string(input_str)
   input_str = 'inputx:file'
   with self.assertRaises(RuntimeError):
     saved_model_cli.preprocess_inputs_arg_string(input_str)
   input_str = 'inputx:np.zeros((5))'
   with self.assertRaises(RuntimeError):
     saved_model_cli.preprocess_input_exprs_arg_string(input_str)
Ejemplo n.º 2
0
 def testInputPreProcessErrorBadFormat(self):
   input_str = 'inputx=file[[v1]v2'
   with self.assertRaises(RuntimeError):
     saved_model_cli.preprocess_inputs_arg_string(input_str)
   input_str = 'inputx:file'
   with self.assertRaises(RuntimeError):
     saved_model_cli.preprocess_inputs_arg_string(input_str)
   input_str = 'inputx:np.zeros((5))'
   with self.assertRaises(RuntimeError):
     saved_model_cli.preprocess_input_exprs_arg_string(input_str)
Ejemplo n.º 3
0
 def testInputPreProcessFileNames(self):
   input_str = (r'inputx=C:\Program Files\data.npz[v:0];'
                r'input:0=c:\PROGRA~1\data.npy')
   input_dict = saved_model_cli.preprocess_inputs_arg_string(input_str)
   self.assertTrue(input_dict['inputx'] == (r'C:\Program Files\data.npz',
                                            'v:0'))
   self.assertTrue(input_dict['input:0'] == (r'c:\PROGRA~1\data.npy', None))
Ejemplo n.º 4
0
 def testInputPreProcessFileNames(self):
   input_str = (r'inputx=C:\Program Files\data.npz[v:0];'
                r'input:0=c:\PROGRA~1\data.npy')
   input_dict = saved_model_cli.preprocess_inputs_arg_string(input_str)
   self.assertTrue(input_dict['inputx'] == (r'C:\Program Files\data.npz',
                                            'v:0'))
   self.assertTrue(input_dict['input:0'] == (r'c:\PROGRA~1\data.npy', None))
Ejemplo n.º 5
0
 def testInputPreProcessFormats(self):
     input_str = 'input1=/path/file.txt[ab3];input2=file2'
     input_expr_str = 'input3=np.zeros([2,2]);input4=[4,5]'
     input_dict = saved_model_cli.preprocess_inputs_arg_string(input_str)
     input_expr_dict = saved_model_cli.preprocess_input_exprs_arg_string(
         input_expr_str)
     self.assertTrue(input_dict['input1'] == ('/path/file.txt', 'ab3'))
     self.assertTrue(input_dict['input2'] == ('file2', None))
     self.assertTrue(input_expr_dict['input3'] == 'np.zeros([2,2])')
     self.assertTrue(input_expr_dict['input4'] == '[4,5]')
     self.assertTrue(len(input_dict) == 2)
     self.assertTrue(len(input_expr_dict) == 2)
Ejemplo n.º 6
0
 def testInputPreProcessFormats(self):
   input_str = 'input1=/path/file.txt[ab3];input2=file2'
   input_expr_str = 'input3=np.zeros([2,2]);input4=[4,5]'
   input_dict = saved_model_cli.preprocess_inputs_arg_string(input_str)
   input_expr_dict = saved_model_cli.preprocess_input_exprs_arg_string(
       input_expr_str)
   self.assertTrue(input_dict['input1'] == ('/path/file.txt', 'ab3'))
   self.assertTrue(input_dict['input2'] == ('file2', None))
   self.assertTrue(input_expr_dict['input3'] == 'np.zeros([2,2])')
   self.assertTrue(input_expr_dict['input4'] == '[4,5]')
   self.assertTrue(len(input_dict) == 2)
   self.assertTrue(len(input_expr_dict) == 2)