Added tests

Author: arthurlw

pandas/tests/arrays/categorical/test_astype.py

@@ -7,14 +7,17 @@
     CategoricalDtype,
     CategoricalIndex,
     DatetimeIndex,
+    Index,
     Interval,
     NaT,
     Period,
     Timestamp,
     array,
+    isna,
     to_datetime,
 )
 import pandas._testing as tm
+from pandas.core.arrays.arrow.array import ArrowExtensionArray
 
 
 class TestAstype:
@@ -160,3 +163,18 @@ def test_astype_category_readonly_mask_values(self):
         result = arr.astype("category")
         expected = array([0, 1, 2], dtype="Int64").astype("category")
         tm.assert_extension_array_equal(result, expected)
+
+    def test_arrow_array_astype_to_categorical_dtype_temporal(self):
+        arr = array(
+            ["2017-01-01", "2018-01-01", "2019-01-01"], dtype="date32[day][pyarrow]"
+        )
+        cats = Index(["2017-01-01", "2018-01-01", "2019-01-01"], dtype="M8[s]")
+        dtype = CategoricalDtype(categories=cats, ordered=False)
+
+        assert not all(isna(arr.astype(dtype)))
+
+        arr = ArrowExtensionArray._from_sequence(["1h", "2h", "3h"])
+        cats = Index(["1h", "2h", "3h"], dtype="m8[ns]")
+        dtype = CategoricalDtype(cats, ordered=False)
+
+        assert not all(isna(arr.astype(dtype)))