def testExecutorModuleFileNotProvided(self): input_dict = {} output = standard_artifacts.DataView() output.uri = os.path.join(self._output_data_dir, 'output_data_view') output_dict = {'data_view': output} exec_properties = { 'create_decoder_func': '%s.%s' % (data_view_module.create_simple_decoder.__module__, data_view_module.create_simple_decoder.__name__), } executor = provider_executor.TfGraphDataViewProviderExecutor() executor.Do(input_dict, output_dict, exec_properties) loaded_decoder = tf_graph_record_decoder.load_decoder(output.uri) self.assertIsInstance(loaded_decoder, tf_graph_record_decoder.TFGraphRecordDecoder)
def testExecutorModuleFileProvided(self): input_dict = {} output = standard_artifacts.DataView() output.uri = os.path.join(self._output_data_dir, 'output_data_view') output_dict = {'data_view': [output]} exec_properties = { 'module_file': os.path.join(self._source_data_dir, 'module_file/data_view_module.py'), 'create_decoder_func': 'create_simple_decoder', } executor = provider_executor.TfGraphDataViewProviderExecutor() executor.Do(input_dict, output_dict, exec_properties) loaded_decoder = tf_graph_record_decoder.load_decoder(output.uri) self.assertIsInstance( loaded_decoder, tf_graph_record_decoder.LoadedDecoder)