(
op: Callable, expected_values: list[object], expected_str: str
)
| 88 | ], |
| 89 | ) |
| 90 | def test_inplace_ops( |
| 91 | op: Callable, expected_values: list[object], expected_str: str |
| 92 | ) -> None: |
| 93 | # https://github.com/pandas-dev/pandas/pull/64267 |
| 94 | df = pd.DataFrame({"a": [1, 2]}) |
| 95 | expr = pd.col("a") |
| 96 | expr = op(expr, 2) |
| 97 | result = df.assign(c=expr) |
| 98 | expected = pd.DataFrame({"a": [1, 2], "c": expected_values}) |
| 99 | tm.assert_frame_equal(result, expected) |
| 100 | assert str(expr) == expected_str |
| 101 | |
| 102 | |
| 103 | def test_matmul(): |