def is_interval_dtype(arr_or_dtype): """ Check whether an array-like or dtype is of the Interval dtype. Parameters ---------- arr_or_dtype : array-like The array-like or dtype to check. Returns ------- boolean Whether or not the array-like or dtype is of the Interval dtype. Examples -------- >>> is_interval_dtype(object) False >>> is_interval_dtype(IntervalDtype()) True >>> is_interval_dtype([1, 2, 3]) False >>> >>> interval = pd.Interval(1, 2, closed="right") >>> is_interval_dtype(interval) False >>> is_interval_dtype(pd.IntervalIndex([interval])) True """ # TODO: Consider making Interval an instance of IntervalDtype if arr_or_dtype is None: return False return IntervalDtype.is_dtype(arr_or_dtype)
def is_interval_dtype(arr_or_dtype) -> bool: """ Check whether an array-like or dtype is of the Interval dtype. Parameters ---------- arr_or_dtype : array-like The array-like or dtype to check. Returns ------- boolean Whether or not the array-like or dtype is of the Interval dtype. Examples -------- >>> is_interval_dtype(object) False >>> is_interval_dtype(IntervalDtype()) True >>> is_interval_dtype([1, 2, 3]) False >>> >>> interval = pd.Interval(1, 2, closed="right") >>> is_interval_dtype(interval) False >>> is_interval_dtype(pd.IntervalIndex([interval])) True """ if isinstance(arr_or_dtype, ExtensionDtype): # GH#33400 fastpath for dtype object return arr_or_dtype.type is Interval if arr_or_dtype is None: return False return IntervalDtype.is_dtype(arr_or_dtype)
def test_is_dtype(self): assert IntervalDtype.is_dtype(self.dtype) assert IntervalDtype.is_dtype('interval') assert IntervalDtype.is_dtype(IntervalDtype('float64')) assert IntervalDtype.is_dtype(IntervalDtype('int64')) assert IntervalDtype.is_dtype(IntervalDtype(np.int64)) assert not IntervalDtype.is_dtype('D') assert not IntervalDtype.is_dtype('3D') assert not IntervalDtype.is_dtype('U') assert not IntervalDtype.is_dtype('S') assert not IntervalDtype.is_dtype('foo') assert not IntervalDtype.is_dtype(np.object_) assert not IntervalDtype.is_dtype(np.int64) assert not IntervalDtype.is_dtype(np.float64)
def test_is_dtype(self): assert IntervalDtype.is_dtype(self.dtype) assert IntervalDtype.is_dtype('interval') assert IntervalDtype.is_dtype(IntervalDtype('float64')) assert IntervalDtype.is_dtype(IntervalDtype('int64')) assert IntervalDtype.is_dtype(IntervalDtype(np.int64)) assert not IntervalDtype.is_dtype('D') assert not IntervalDtype.is_dtype('3D') assert not IntervalDtype.is_dtype('U') assert not IntervalDtype.is_dtype('S') assert not IntervalDtype.is_dtype('foo') assert not IntervalDtype.is_dtype(np.object_) assert not IntervalDtype.is_dtype(np.int64) assert not IntervalDtype.is_dtype(np.float64)
def test_is_dtype(self): assert IntervalDtype.is_dtype(self.dtype) assert IntervalDtype.is_dtype("interval") assert IntervalDtype.is_dtype(IntervalDtype("float64")) assert IntervalDtype.is_dtype(IntervalDtype("int64")) assert IntervalDtype.is_dtype(IntervalDtype(np.int64)) assert not IntervalDtype.is_dtype("D") assert not IntervalDtype.is_dtype("3D") assert not IntervalDtype.is_dtype("U") assert not IntervalDtype.is_dtype("S") assert not IntervalDtype.is_dtype("foo") assert not IntervalDtype.is_dtype("IntervalA") assert not IntervalDtype.is_dtype(np.object_) assert not IntervalDtype.is_dtype(np.int64) assert not IntervalDtype.is_dtype(np.float64)
def test_is_dtype(self): self.assertTrue(IntervalDtype.is_dtype(self.dtype)) self.assertTrue(IntervalDtype.is_dtype('interval')) self.assertTrue(IntervalDtype.is_dtype(IntervalDtype('float64'))) self.assertTrue(IntervalDtype.is_dtype(IntervalDtype('int64'))) self.assertTrue(IntervalDtype.is_dtype(IntervalDtype(np.int64))) self.assertFalse(IntervalDtype.is_dtype('D')) self.assertFalse(IntervalDtype.is_dtype('3D')) self.assertFalse(IntervalDtype.is_dtype('U')) self.assertFalse(IntervalDtype.is_dtype('S')) self.assertFalse(IntervalDtype.is_dtype('foo')) self.assertFalse(IntervalDtype.is_dtype(np.object_)) self.assertFalse(IntervalDtype.is_dtype(np.int64)) self.assertFalse(IntervalDtype.is_dtype(np.float64))
def test_is_dtype(self, dtype): assert IntervalDtype.is_dtype(dtype) assert IntervalDtype.is_dtype("interval") assert IntervalDtype.is_dtype(IntervalDtype("float64")) assert IntervalDtype.is_dtype(IntervalDtype("int64")) assert IntervalDtype.is_dtype(IntervalDtype(np.int64)) assert IntervalDtype.is_dtype(IntervalDtype("float64", "left")) assert IntervalDtype.is_dtype(IntervalDtype("int64", "right")) assert IntervalDtype.is_dtype(IntervalDtype(np.int64, "both")) assert not IntervalDtype.is_dtype("D") assert not IntervalDtype.is_dtype("3D") assert not IntervalDtype.is_dtype("U") assert not IntervalDtype.is_dtype("S") assert not IntervalDtype.is_dtype("foo") assert not IntervalDtype.is_dtype("IntervalA") assert not IntervalDtype.is_dtype(np.object_) assert not IntervalDtype.is_dtype(np.int64) assert not IntervalDtype.is_dtype(np.float64)
def test_is_dtype(self): self.assertTrue(IntervalDtype.is_dtype(self.dtype)) self.assertTrue(IntervalDtype.is_dtype('interval')) self.assertTrue(IntervalDtype.is_dtype(IntervalDtype('float64'))) self.assertTrue(IntervalDtype.is_dtype(IntervalDtype('int64'))) self.assertTrue(IntervalDtype.is_dtype(IntervalDtype(np.int64))) self.assertFalse(IntervalDtype.is_dtype('D')) self.assertFalse(IntervalDtype.is_dtype('3D')) self.assertFalse(IntervalDtype.is_dtype('U')) self.assertFalse(IntervalDtype.is_dtype('S')) self.assertFalse(IntervalDtype.is_dtype('foo')) self.assertFalse(IntervalDtype.is_dtype(np.object_)) self.assertFalse(IntervalDtype.is_dtype(np.int64)) self.assertFalse(IntervalDtype.is_dtype(np.float64))