def test_series_set_equal_length_object_by_mask(replace_data):
psr = pd.Series([1, 2, 3, 4, 5])
gsr = Series.from_pandas(psr)
# Lengths match in trivial case
pd_bool_col = pd.Series([True] * len(psr))
gd_bool_col = Series.from_pandas(pd_bool_col)
psr[pd_bool_col] = (
replace_data.to_pandas()
if hasattr(replace_data, "to_pandas")
else replace_data
)
gsr[gd_bool_col] = replace_data
assert_eq(psr.astype("float"), gsr.astype("float"))
# Test partial masking
psr[psr > 1] = (
replace_data.to_pandas()
if hasattr(replace_data, "to_pandas")
else replace_data
)
gsr[gsr > 1] = replace_data
assert_eq(psr.astype("float"), gsr.astype("float"))
def test_series_set_item(psr, arg):
gsr = Series.from_pandas(psr)
psr[arg] = 11
gsr[arg] = 11
assert_eq(psr, gsr)
