BUG/REF: rewrite write/write_array to handle tz info for series

Author: arw2019

pandas/io/pytables.py

@@ -2960,11 +2960,17 @@ def write_array_empty(self, key: str, value: ArrayLike):
         node._v_attrs.value_type = str(value.dtype)
         node._v_attrs.shape = value.shape
 
-    def write_array(self, key: str, value: ArrayLike, items: Optional[Index] = None):
+    def write_array(self, key: str, obj, items: Optional[Index] = None):
         # TODO: we only have one test that gets here, the only EA
         #  that gets passed is DatetimeArray, and we never have
         #  both self._filters and EA
-        assert isinstance(value, (np.ndarray, ABCExtensionArray)), type(value)
+
+        if isinstance(obj, (np.ndarray, ABCExtensionArray)):
+            value = obj
+        elif is_datetime64tz_dtype(obj):
+            value = obj.dt._get_values()
+        else:
+            value = obj.values
 
         if key in self.group:
             self._handle.remove_node(self.group, key)
@@ -3019,8 +3025,6 @@ def write_array(self, key: str, value: ArrayLike, items: Optional[Index] = None)
             vlarr = self._handle.create_vlarray(self.group, key, _tables().ObjectAtom())
             vlarr.append(value)
 
-        elif empty_array:
-            self.write_array_empty(key, value)
         elif is_datetime64_dtype(value.dtype):
             self._handle.create_array(self.group, key, value.view("i8"))
             getattr(self.group, key)._v_attrs.value_type = "datetime64"
@@ -3035,6 +3039,8 @@ def write_array(self, key: str, value: ArrayLike, items: Optional[Index] = None)
         elif is_timedelta64_dtype(value.dtype):
             self._handle.create_array(self.group, key, value.view("i8"))
             getattr(self.group, key)._v_attrs.value_type = "timedelta64"
+        elif empty_array:
+            self.write_array_empty(key, value)
         else:
             self._handle.create_array(self.group, key, value)
 
@@ -3069,7 +3075,7 @@ def read(
     def write(self, obj, **kwargs):
         super().write(obj, **kwargs)
         self.write_index("index", obj.index)
-        self.write_array("values", obj.values)
+        self.write_array("values", obj)
         self.attrs.name = obj.name
 
 
@@ -4728,13 +4734,8 @@ def _set_tz(
         assert values.tz is None or values.tz == tz
 
     if tz is not None:
-        if isinstance(values, DatetimeIndex):
-            name = values.name
-            values = values.asi8
-        else:
-            name = None
-            values = values.ravel()
-
+        name = getattr(values, "name", None)
+        values = values.ravel()
         tz = _ensure_decoded(tz)
         values = DatetimeIndex(values, name=name)
         values = values.tz_localize("UTC").tz_convert(tz)