Fix DatetimeArray-case in test_datetime64tz_aware; create TODO

Author: h-vetinari

pandas/tests/test_algos.py

@@ -37,9 +37,12 @@ def assert_series_or_index_or_array_or_categorical_equal(left, right):
  tm.assert_numpy_array_equal(left, right)
  elif isinstance(left, Categorical):
  tm.assert_categorical_equal(left, right)
+ elif isinstance(left, DatetimeArray):
+ tm.assert_extension_array_equal(left, right)
  else:
  # will fail
- assert isinstance(left, (Series, Index, np.ndarray, Categorical))
+ assert isinstance(left, (Series, Index, np.ndarray,
+ Categorical, DatetimeArray))
 
 
 class TestMatch(object):
@@ -493,19 +496,19 @@ def test_datetime64tz_aware(self, method, box):
  result = method(obj)
  assert_series_or_index_or_array_or_categorical_equal(result, expected)
 
- if method == pd.unique:
-  # [Series/Index].unique do not yet support return_inverse=True
-
-   # reuse result as expected outcome of return_inverse case
-   expected_uniques = result.copy()
-   result_uniques, result_inverse = method(obj, return_inverse=True)
-
-   assert_series_or_index_or_array_or_categorical_equal(
-   result_uniques, expected_uniques)
-
-   # reconstruction can only work if inverse is correct
-   reconstr = box(result_uniques[result_inverse])
-   assert_series_or_index_or_array_or_categorical_equal(reconstr, obj)
+ # TODO: add support for return_inverse to DatetimeArray/DatetimeIndex,
+ # as well as [[Series/Index].unique
+
+ # # reuse result as expected outcome of return_inverse case
+ # expected_uniques = result.copy()
+ # result_uniques, result_inverse = method(obj, return_inverse=True)
+ #
+ # assert_series_or_index_or_array_or_categorical_equal(
+ # result_uniques, expected_uniques)
+ #
+ # # reconstruction can only work if inverse is correct
+ # reconstr = box(result_uniques[result_inverse])
+ # assert_series_or_index_or_array_or_categorical_equal(reconstr, obj)
 
  def test_order_of_appearance(self):
  # 9346