def test_do_without_blessed_model(self): # Create exe properties. exec_properties = { 'blessed_model': None, 'blessed_model_id': None, } # Run executor. model_validator = executor.Executor() model_validator.Do(self._input_dict, self._output_dict, exec_properties) # Check model validator outputs. self.assertTrue( tf.gfile.Exists(os.path.join(self._blessing.uri, 'BLESSED')))
def testDoWithoutBlessedModel(self): # Create exe properties. exec_properties = { 'blessed_model': None, 'blessed_model_id': None, 'component_id': self.component_id, } # Run executor. model_validator = executor.Executor(self._context) model_validator.Do(self._input_dict, self._output_dict, exec_properties) # Check model validator outputs. self.assertTrue(tf.io.gfile.exists(os.path.join(self._tmp_dir))) self.assertTrue( tf.io.gfile.exists(os.path.join(self._blessing.uri, 'BLESSED')))
def testDoWithBlessedModel(self): # Create exe properties. exec_properties = { 'blessed_model': os.path.join(self._source_data_dir, 'trainer/blessed'), 'blessed_model_id': 123, 'current_component_id': self.component_id, } # Run executor. model_validator = executor.Executor(self._context) model_validator.Do(self._input_dict, self._output_dict, exec_properties) # Check model validator outputs. self.assertTrue(fileio.exists(os.path.join(self._tmp_dir))) self.assertTrue( fileio.exists( os.path.join(self._blessing.uri, constants.BLESSED_FILE_NAME)))
def test_do_with_blessed_model(self): # Create exe properties. exec_properties = { 'blessed_model': os.path.join(self._source_data_dir, 'trainer/blessed/'), 'blessed_model_id': 123, } # Run executor. model_validator = executor.Executor(self._context) model_validator.Do(self._input_dict, self._output_dict, exec_properties) # Check model validator outputs. self.assertTrue( tf.gfile.Exists(os.path.join(self._tmp_dir))) self.assertTrue( tf.gfile.Exists(os.path.join(self._blessing.uri, 'BLESSED')))