Move Timestamp dtype to test_ file

Author: xhochy

pandas/tests/extension/arrow/arrays.py

@@ -7,7 +7,6 @@
 current implementation is not efficient.
 """
 import copy
-import datetime
 import itertools
 import operator
 from typing import Type
@@ -68,26 +67,6 @@ def construct_array_type(cls) -> Type["ArrowStringArray"]:
         return ArrowStringArray
 
 
-@register_extension_dtype
-class ArrowTimestampUSDtype(ExtensionDtype):
-
-    type = datetime.datetime
-    kind = "M"
-    name = "arrow_timestamp_us"
-    na_value = pa.NULL
-
-    @classmethod
-    def construct_array_type(cls) -> Type["ArrowTimestampUSArray"]:
-        """
-        Return the array type associated with this dtype.
-
-        Returns
-        -------
-        type
-        """
-        return ArrowTimestampUSArray
-
-
 class ArrowExtensionArray(ExtensionArray):
     @classmethod
     def from_scalars(cls, values):
@@ -222,13 +201,3 @@ def __init__(self, values):
         assert values.type == pa.string()
         self._data = values
         self._dtype = ArrowStringDtype()
-
-
-class ArrowTimestampUSArray(ArrowExtensionArray):
-    def __init__(self, values):
-        if not isinstance(values, pa.ChunkedArray):
-            raise ValueError
-
-        assert values.type == pa.timestamp("us")
-        self._data = values
-        self._dtype = ArrowTimestampUSDtype()

pandas/tests/extension/arrow/test_timestamp.py

@@ -1,12 +1,46 @@
 import datetime
+from typing import Type
 
 import pytest
 
 import pandas as pd
+from pandas.api.extensions import ExtensionDtype, register_extension_dtype
 
 pytest.importorskip("pyarrow", minversion="0.13.0")
 
-from .arrays import ArrowTimestampUSArray  # isort:skip
+import pyarrow as pa  # isort:skip
+
+from .arrays import ArrowExtensionArray  # isort:skip
+
+
+@register_extension_dtype
+class ArrowTimestampUSDtype(ExtensionDtype):
+
+    type = datetime.datetime
+    kind = "M"
+    name = "arrow_timestamp_us"
+    na_value = pa.NULL
+
+    @classmethod
+    def construct_array_type(cls) -> Type["ArrowTimestampUSArray"]:
+        """
+        Return the array type associated with this dtype.
+
+        Returns
+        -------
+        type
+        """
+        return ArrowTimestampUSArray
+
+
+class ArrowTimestampUSArray(ArrowExtensionArray):
+    def __init__(self, values):
+        if not isinstance(values, pa.ChunkedArray):
+            raise ValueError
+
+        assert values.type == pa.timestamp("us")
+        self._data = values
+        self._dtype = ArrowTimestampUSDtype()
 
 
 @pytest.mark.xfail(