test

Author: andrewgsavage

pandas/tests/dtypes/test_inference.py

@@ -75,19 +75,14 @@
     FloatingArray,
     IntegerArray,
 )
+from pandas.core.dtypes.dtypes import ExtensionDtype
 
+from pandas.core
 
 @pytest.fixture(params=[True, False], ids=str)
 def coerce(request):
     return request.param
 
-class MockScalar:
-    pass
-
-class MockDtype(pd.api.extensions.ExtensionDtype):
-    
-
-
 
 class MockNumpyLikeArray:
     """
@@ -2032,3 +2027,18 @@ def test_find_result_type_int_int(right, result):
 def test_find_result_type_floats(right, result):
     left_dtype = np.dtype("float16")
     assert find_result_type(left_dtype, right) == result
+
+def test_infer_dtype_extensiondtype():
+    class MockScalar:
+        pass
+
+    class MockDtype(ExtensionDtype):
+        @property
+        def name(self):
+            return "MockDtype"
+        def is_unambiguous_scalar(scalar):
+            if isinstance(scalar, MockScalar):
+                return True
+            return False
+    arr = [MockScalar()]
+    assert lib.infer_dtype(arr, skipna=True) == "MockDtype"