Add tests for NaT when performing dt.to_period (pandas-dev#24921)

Pingviinituutti · Feb 28, 2019 · 60fab6e · 60fab6e
1 parent c539848
commit 60fab6e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pandas/tests/series/test_period.py b/pandas/tests/series/test_period.py
@@ -164,3 +164,12 @@ def test_end_time_timevalues(self, input_vals):
         result = s.dt.end_time
         expected = s.apply(lambda x: x.end_time)
         tm.assert_series_equal(result, expected)
+
+    @pytest.mark.parametrize('input_vals', [
+        ('2001'), ('NaT')
+    ])
+    def test_to_period(self, input_vals):
+        # GH 21205
+        expected = Series([input_vals], dtype='Period[D]')
+        result = Series([input_vals], dtype='datetime64[ns]').dt.to_period('D')
+        tm.assert_series_equal(result, expected)