from __future__ import annotations

import numpy as np
import pandas as pd

import dask.dataframe.methods as methods
from dask.dataframe._compat import PANDAS_GE_140


def test_assign_not_modifying_array_inplace():
    df = pd.DataFrame({"a": [1, 2, 3], "b": 1.5})
    result = methods.assign(df, "a", 5)
    assert not np.shares_memory(df["a"].values, result["a"].values)
    if PANDAS_GE_140:
        assert np.shares_memory(df["b"].values, result["b"].values)