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(