def load_pipeline(filename): """Loads a single pipeline from a filename. """ initialize() if not isinstance(filename, basestring): raise TypeError("load_vistrails() expects a filename, got %r" % type(filename).__name__) # Copied from VistrailsApplicationInterface#open_workflow() locator = FileLocator(filename) pipeline = locator.load(_Pipeline) return Pipeline(pipeline)
def do_export(self, filename, expected_source): import os from vistrails.core.db.locator import FileLocator from vistrails.core.system import vistrails_root_directory from vistrails.core.vistrail.pipeline import Pipeline locator = FileLocator( os.path.join(vistrails_root_directory(), 'tests', 'resources', filename)) pipeline = locator.load(Pipeline) self.assertEqual('\n'.join(write_workflow_to_python(pipeline)) + '\n', expected_source)