ray/streaming/python/tests/test_operator.py
2020-02-25 10:33:33 +08:00

8 lines
263 B
Python

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