snowpark-02.py

def test_filter(session: Session):
    source_data = [...]
    schema = T.StructType(...)
    source_df = session.create_dataframe(source_data, schema=schema)
    actual_df = process.filter_personal_consumption_expenditures(source_df)
    expected_data = [(2021, 115.53)]
    expected_df = session.create_dataframe(
        expected_data,
        schema=T.StructType(
            [
                T.StructField("Year", T.IntegerType(), nullable=True),
                T.StructField("PCE", T.DoubleType(), nullable=True),
            ]
        ),
    )
    assert actual_df.collect() == expected_df.collect()