Function test_groupby

python/pathway/tests/test_py_object_wrapper.py:132–159

()

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

    t = pw.debug.table_from_markdown(
        """
        a
        1
        2
        2
        3
        1
    """
    ).select(simple=create_simple(pw.this.a))

    res = t.groupby(pw.this.simple).reduce(cnt=pw.reducers.count())

    expected = pw.debug.table_from_markdown(
        """
        cnt
        2
        2
        1
    """
    )

    assert_table_equality_wo_index(res, expected)


def test_compute_and_print():