move test

tm9k1 · Oct 2, 2018 · be63feb · be63feb
1 parent 0eef0cf
commit be63feb
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 20 deletions.
diff --git a/pandas/tests/extension/decimal/decimal_array/test_decimal.py b/pandas/tests/extension/decimal/decimal_array/test_decimal.py
@@ -1,3 +1,4 @@
+import operator
 import decimal
 
 import random
@@ -275,3 +276,19 @@ def test_compare_array(self, data, all_compare_operators):
         other = pd.Series(data) * [decimal.Decimal(pow(2.0, i))
                                    for i in alter]
         self._compare_other(s, data, op_name, other)
+
+
+def test_combine_from_sequence_raises():
+    # https://github.com/pandas-dev/pandas/issues/22850
+    class BadDecimalArray(DecimalArray):
+        def _from_sequence(cls, scalars, dtype=None, copy=False):
+            raise KeyError("For the test")
+
+    ser = pd.Series(BadDecimalArray([decimal.Decimal("1.0"),
+                                     decimal.Decimal("2.0")]))
+    result = ser.combine(ser, operator.add)
+
+    # note: object dtype
+    expected = pd.Series([decimal.Decimal("2.0"),
+                          decimal.Decimal("4.0")], dtype="object")
+    tm.assert_series_equal(result, expected)
diff --git a/pandas/tests/extension/test_ops.py b/pandas/tests/extension/test_ops.py