1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
| from pandas import (
| DataFrame,
| Series,
| )
| import pandas._testing as tm
|
|
| class TestDataFrameSetItem:
| def test_isetitem_ea_df(self):
| # GH#49922
| df = DataFrame([[1, 2, 3], [4, 5, 6]])
| rhs = DataFrame([[11, 12], [13, 14]], dtype="Int64")
|
| df.isetitem([0, 1], rhs)
| expected = DataFrame(
| {
| 0: Series([11, 13], dtype="Int64"),
| 1: Series([12, 14], dtype="Int64"),
| 2: [3, 6],
| }
| )
| tm.assert_frame_equal(df, expected)
|
| def test_isetitem_ea_df_scalar_indexer(self):
| # GH#49922
| df = DataFrame([[1, 2, 3], [4, 5, 6]])
| rhs = DataFrame([[11], [13]], dtype="Int64")
|
| df.isetitem(2, rhs)
| expected = DataFrame(
| {
| 0: [1, 4],
| 1: [2, 5],
| 2: Series([11, 13], dtype="Int64"),
| }
| )
| tm.assert_frame_equal(df, expected)
|
|
