ray/streaming/python/tests/test_operator.py

9 lines
263 B
Python
Raw Normal View History

from ray.streaming import operator
from ray.streaming import function
def test_create_operator():
map_func = function.SimpleMapFunction(lambda x: x)
map_operator = operator.create_operator(map_func)
assert type(map_operator) is operator.MapOperator