From dc17ee7641aeb98a5064bda0407dff10a0a7e2de Mon Sep 17 00:00:00 2001 From: Daniel Saxton Date: Wed, 6 May 2020 11:52:51 -0500 Subject: [PATCH] TST: Add Series.update ExtensionArray tests --- pandas/tests/series/methods/test_update.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/pandas/tests/series/methods/test_update.py b/pandas/tests/series/methods/test_update.py index 9cb6d8f81fff7..d00a4299cb690 100644 --- a/pandas/tests/series/methods/test_update.py +++ b/pandas/tests/series/methods/test_update.py @@ -1,7 +1,7 @@ import numpy as np import pytest -from pandas import DataFrame, Series +from pandas import CategoricalDtype, DataFrame, NaT, Series, Timestamp import pandas._testing as tm @@ -93,6 +93,16 @@ def test_update_from_non_series(self, series, other, expected): Series([None, False], dtype="boolean"), Series([True, False], dtype="boolean"), ), + ( + Series(["a", None], dtype=CategoricalDtype(categories=["a", "b"])), + Series([None, "b"], dtype=CategoricalDtype(categories=["a", "b"])), + Series(["a", "b"], dtype=CategoricalDtype(categories=["a", "b"])), + ), + ( + Series([Timestamp(year=2020, month=1, day=1, tz="Europe/London"), NaT]), + Series([NaT, Timestamp(year=2020, month=1, day=1, tz="Europe/London")]), + Series([Timestamp(year=2020, month=1, day=1, tz="Europe/London")] * 2), + ), ], ) def test_update_extension_array_series(self, result, target, expected):