def test_call_function(self):
func_name = 'marmot.preprocessing.parsers.parse_corpus_contexts'
func = import_utils.import_function(func_name)
interesting_tokens = set(['the', 'it'])
corpus_path = os.path.join(self.module_path, 'test_data/corpus.en.1000')
args = [corpus_path, interesting_tokens]
result = import_utils.call_function(func, args)
self.assertTrue(len(result) > 0)
def test_call_sample_function_tree(self):
graph_obj = self.config['test_obj']
func = import_utils.import_function(graph_obj['func'])
args = graph_obj['args']
json_contexts = import_utils.function_tree(func, args)
self.assertTrue(type(json_contexts) == str)
self.assertListEqual(json.loads(json_contexts), self.contexts)
def test_import_function(self):
func_name = 'marmot.preprocessing.parsers.parse_back_translation'
func = import_utils.import_function(func_name)
self.assertTrue(inspect.isfunction(func))
