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

alexander-ponomaroff · Pingviinituutti · commit 9f0856472970 · 2019-02-28T10:26:56.000+02:00
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)
