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')))
