Function test_interchange_protocol

dask/dataframe/dask_expr/tests/test_interchange.py:12–20

()

def test_interchange_protocol():
    pdf = pd.DataFrame({"a": [1, 2, 3], "b": 1})
    df = from_pandas(pdf, npartitions=2)
    df_int = df.__dataframe__()
    pd.testing.assert_index_equal(pdf.columns, df_int.column_names())
    assert df_int.num_columns() == 2
    assert df_int.num_rows() == 3
    column = df_int.get_columns()[0]
    assert column.dtype()[0] == DtypeKind.INT