ray/streaming/python/tests/test_operator.py

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