Function test_py_object_simple

python/pathway/tests/test_py_object_wrapper.py:35–64

()

def test_py_object_simple():

    @pw.udf
    def create_py_object(a: int) -> pw.PyObjectWrapper[Simple]:
        return pw.PyObjectWrapper(Simple(a))

    @pw.udf
    def use_py_object(a: int, b: pw.PyObjectWrapper[Simple]) -> int:
        return b.value.add(a)

    t = pw.debug.table_from_markdown(
        """
        a
        1
        2
        3
    """
    ).with_columns(b=create_py_object(pw.this.a))

    res = t.select(res=use_py_object(pw.this.a, pw.this.b))

    expected = pw.debug.table_from_markdown(
        """
        res
        2
        4
        6
    """
    )
    assert_table_equality(res, expected)


@dataclass