Python infer_output_props Examples

Example #1

File: test_inference.py Project: jordrop/dagster

def test_infer_output_description_from_docstring_rest():
    @solid
    def rest(_context) -> int:
        """
        :return int: a number
        """

    props = infer_output_props(rest.compute_fn)
    assert props.description == "a number"
    assert props.annotation == int

Example #2

def test_infer_output_description_from_docstring_google():
    @solid
    def google(_context) -> int:
        """
        Returns:
            int: a number
        """

    props = infer_output_props(google.compute_fn.decorated_fn)

    assert props.description == "a number"
    assert props.annotation == int

Example #3

def test_infer_output_description_from_docstring_numpy():
    @solid
    def numpy(_context) -> int:
        """

        Returns
        -------
        int
            a number
        """

    props = infer_output_props(numpy.compute_fn.decorated_fn)
    assert props.description == "a number"
    assert props.annotation == int