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)