def test_register():
from vital.config import config
from vital.modules import modules
from sprokit.pipeline import pipeline
from sprokit.pipeline import scheduler_factory
modules.load_known_modules()
sched_type = 'python_example'
sched_desc = 'simple description'
scheduler_factory.add_scheduler(sched_type, sched_desc,
example_scheduler(True))
if not sched_desc == scheduler_factory.description(sched_type):
test_error("Description was not preserved when registering")
p = pipeline.Pipeline()
try:
s = scheduler_factory.create_scheduler(sched_type, p)
if s is None:
raise Exception()
except:
test_error("Could not create newly registered scheduler type")
def test_api_calls():
from vital.config import config
from vital.modules import modules
from sprokit.pipeline import pipeline
from sprokit.pipeline import scheduler_factory
modules.load_known_modules()
sched_type = 'thread_per_process'
c = config.empty_config()
p = pipeline.Pipeline()
scheduler_factory.create_scheduler(sched_type, p)
scheduler_factory.create_scheduler(sched_type, p, c)
scheduler_factory.types()
scheduler_factory.description(sched_type)
scheduler_factory.default_type
def test_api_calls():
from sprokit.pipeline import config
from vital.modules import modules
from sprokit.pipeline import pipeline
from sprokit.pipeline import scheduler_factory
modules.load_known_modules()
sched_type = 'thread_per_process'
c = config.empty_config()
p = pipeline.Pipeline()
scheduler_factory.create_scheduler(sched_type, p)
scheduler_factory.create_scheduler(sched_type, p, c)
scheduler_factory.types()
scheduler_factory.description(sched_type)
scheduler_factory.default_type
def run_pipeline(sched_type, pipe, conf):
from sprokit.pipeline import config
from sprokit.pipeline import modules
from sprokit.pipeline import scheduler_factory
modules.load_known_modules()
s = scheduler_factory.create_scheduler(sched_type, pipe, conf)
s.start()
s.wait()
def run_pipeline(sched_type, pipe, conf):
from vital.config import config
from vital.modules import modules
from sprokit.pipeline import scheduler_factory
import sys
modules.load_known_modules()
if sched_type in cpp_scheds:
expect_exception("trying to run a python process on a C++ scheduler", RuntimeError,
scheduler_factory.create_scheduler, sched_type, pipe, conf)
else:
s = scheduler_factory.create_scheduler(sched_type, pipe, conf)
s.start()
s.wait()
def run_pipeline(sched_type, pipe, conf):
from sprokit.pipeline import config
from sprokit.pipeline import modules
from sprokit.pipeline import scheduler_factory
import sys
modules.load_known_modules()
if sched_type in cpp_scheds:
expect_exception("trying to run a python process on a C++ scheduler", RuntimeError,
scheduler_factory.create_scheduler, sched_type, pipe, conf)
else:
s = scheduler_factory.create_scheduler(sched_type, pipe, conf)
s.start()
s.wait()
def test_wrapper_api():
from vital.config import config
from vital.modules import modules
from sprokit.pipeline import pipeline
from sprokit.pipeline import process_factory
from sprokit.pipeline import scheduler_factory
sched_type = 'python_example'
sched_desc = 'simple description'
modules.load_known_modules()
scheduler_factory.add_scheduler(sched_type, sched_desc,
example_scheduler(False))
p = pipeline.Pipeline()
proc_type = 'orphan'
proc_name = 'orphan'
proc = process_factory.create_process(proc_type, proc_name)
p.add_process(proc)
def check_scheduler(s):
if s is None:
test_error("Got a 'None' scheduler")
return
s.start()
s.pause()
s.resume()
s.stop()
s.start()
s.wait()
del s
p.reset()
p.setup_pipeline()
s = scheduler_factory.create_scheduler(sched_type, p)
check_scheduler(s)
def test_wrapper_api():
from sprokit.pipeline import config
from vital.modules import modules
from sprokit.pipeline import pipeline
from sprokit.pipeline import process_factory
from sprokit.pipeline import scheduler_factory
sched_type = 'python_example'
sched_desc = 'simple description'
modules.load_known_modules()
scheduler_factory.add_scheduler(sched_type, sched_desc, example_scheduler(False))
p = pipeline.Pipeline()
proc_type = 'orphan'
proc_name = 'orphan'
proc = process_factory.create_process(proc_type, proc_name)
p.add_process(proc)
def check_scheduler(s):
if s is None:
test_error("Got a 'None' scheduler")
return
s.start()
s.pause()
s.resume()
s.stop()
s.start()
s.wait()
del s
p.reset()
p.setup_pipeline()
s = scheduler_factory.create_scheduler(sched_type, p)
check_scheduler(s)
def test_register():
from sprokit.pipeline import config
from vital.modules import modules
from sprokit.pipeline import pipeline
from sprokit.pipeline import scheduler_factory
modules.load_known_modules()
sched_type = 'python_example'
sched_desc = 'simple description'
scheduler_factory.add_scheduler(sched_type, sched_desc, example_scheduler(True))
if not sched_desc == scheduler_factory.description(sched_type):
test_error("Description was not preserved when registering")
p = pipeline.Pipeline()
try:
s = scheduler_factory.create_scheduler(sched_type, p)
if s is None:
raise Exception()
except:
test_error("Could not create newly registered scheduler type")