def testDoBlessed(self):
    self._model_blessing.uri = os.path.join(self._source_data_dir, 'model_validator/blessed/')
    self._model_blessing.set_int_custom_property('blessed', 1)
    self._executor.Do(self._input_dict, self._output_dict, self._exec_properties)
    self.assertNotEqual(0, len(tf.io.gfile.listdir(self._serving_model_dir)))
    self.assertNotEqual(0, len(tf.io.gfile.listdir(self._model_push.uri)))
    self.assertEqual(1, self._model_push.artifact.custom_properties['pushed'].int_value)
def testDoBlessed(self):
    self._model_blessing.uri = os.path.join(self._source_data_dir, 'model_validator/blessed/')
    self._model_blessing.set_int_custom_property('blessed', 1)
    self._executor.Do(self._input_dict, self._output_dict, self._exec_properties)
    self.assertNotEqual(0, len(tf.io.gfile.listdir(self._serving_model_dir)))
    self.assertNotEqual(0, len(tf.io.gfile.listdir(self._model_push.uri)))
    self.assertEqual(1, self._model_push.artifact.custom_properties['pushed'].int_value)