| 123456789101112131415161718192021222324252627282930313233343536373839404142434445 |
- from datetime import datetime
- import numpy as np
- from pandas import (
- DatetimeIndex,
- Series,
- )
- import pandas._testing as tm
- def test_series_set_value():
- # GH#1561
- dates = [datetime(2001, 1, 1), datetime(2001, 1, 2)]
- index = DatetimeIndex(dates)
- s = Series(dtype=object)
- s._set_value(dates[0], 1.0)
- s._set_value(dates[1], np.nan)
- expected = Series([1.0, np.nan], index=index)
- tm.assert_series_equal(s, expected)
- def test_set_value_dt64(datetime_series):
- idx = datetime_series.index[10]
- res = datetime_series._set_value(idx, 0)
- assert res is None
- assert datetime_series[idx] == 0
- def test_set_value_str_index(string_series):
- # equiv
- ser = string_series.copy()
- res = ser._set_value("foobar", 0)
- assert res is None
- assert ser.index[-1] == "foobar"
- assert ser["foobar"] == 0
- ser2 = string_series.copy()
- ser2.loc["foobar"] = 0
- assert ser2.index[-1] == "foobar"
- assert ser2["foobar"] == 0
|
