def test(self, **kwargs):
super(JSONFormatTester, self).test(**kwargs)
stream = kwargs.get('stream')
dataformat = kwargs.get('dataformat')
inst = self.cls()
from percept.utils.registry import registry, find_in_registry
selected_registry = find_in_registry(category = RegistryCategories.inputs)
selected_input_registry = [i for i in selected_registry if i.input_format==dataformat]
if len(selected_input_registry)>0:
input_inst = selected_input_registry[0]()
input_inst.read_input(stream)
inst.read_input(input_inst.get_data(), dataformat)
for output_format in inst.output_formats:
data = inst.get_data(output_format)
assert data is not None
def setup_tasks(self, tasks):
"""
Find task classes from category.namespace.name strings
tasks - list of strings
"""
task_classes = []
for task in tasks:
category, namespace, name = task.split(".")
try:
cls = find_in_registry(category=category, namespace=namespace, name=name)[0]
except TypeError:
log.error("Could not find the task with category.namespace.name {0}".format(task))
raise TypeError
task_classes.append(cls)
self.tasks = task_classes
def test(self, **kwargs):
super(JSONFormatTester, self).test(**kwargs)
stream = kwargs.get('stream')
dataformat = kwargs.get('dataformat')
inst = self.cls()
from percept.utils.registry import registry, find_in_registry
selected_registry = find_in_registry(
category=RegistryCategories.inputs)
selected_input_registry = [
i for i in selected_registry if i.input_format == dataformat
]
if len(selected_input_registry) > 0:
input_inst = selected_input_registry[0]()
input_inst.read_input(stream)
inst.read_input(input_inst.get_data(), dataformat)
for output_format in inst.output_formats:
data = inst.get_data(output_format)
assert data is not None
def setup_tasks(self, tasks):
"""
Find task classes from category.namespace.name strings
tasks - list of strings
"""
task_classes = []
for task in tasks:
category, namespace, name = task.split(".")
try:
cls = find_in_registry(category=category,
namespace=namespace,
name=name)[0]
except TypeError:
log.error(
"Could not find the task with category.namespace.name {0}".
format(task))
raise TypeError
task_classes.append(cls)
self.tasks = task_classes
