test_set_name.py 595 B

123456789101112131415161718192021
  1. from datetime import datetime
  2. from pandas import Series
  3. class TestSetName:
  4. def test_set_name(self):
  5. ser = Series([1, 2, 3])
  6. ser2 = ser._set_name("foo")
  7. assert ser2.name == "foo"
  8. assert ser.name is None
  9. assert ser is not ser2
  10. def test_set_name_attribute(self):
  11. ser = Series([1, 2, 3])
  12. ser2 = Series([1, 2, 3], name="bar")
  13. for name in [7, 7.0, "name", datetime(2001, 1, 1), (1,), "\u05D0"]:
  14. ser.name = name
  15. assert ser.name == name
  16. ser2.name = name
  17. assert ser2.name == name