TST: add datetime column to test

Author: arw2019

pandas/tests/frame/methods/test_to_dict.py

@@ -259,15 +259,21 @@ def test_to_dict_wide(self):
     def test_to_dict_orient_dtype(self):
         # GH#22620
         # Input Data
-        input_data = {"a": [1, 2, 3], "b": [1.0, 2.0, 3.0], "c": ["X", "Y", "Z"]}
+        input_data = {
+            "a": [1, 2, 3],
+            "b": [1.0, 2.0, 3.0],
+            "c": ["X", "Y", "Z"],
+            "d": [datetime(2018, 1, 1), datetime(2019, 2, 2), datetime(2020, 3, 3)],
+        }
         df = DataFrame(input_data)
         # Expected Dtypes
-        expected = {"a": int, "b": float, "c": str}
+        expected = {"a": int, "b": float, "c": str, "d": datetime}
         # Extracting dtypes out of to_dict operation
         for df_dict in df.to_dict("records"):
             result = {
                 "a": type(df_dict["a"]),
                 "b": type(df_dict["b"]),
                 "c": type(df_dict["c"]),
+                "d": type(df_dict["d"]),
             }
             assert result == expected