def test_dtype_equal(): assert is_dtype_equal(np.int64, np.int64) assert not is_dtype_equal(np.int64, np.float64) p1 = PeriodDtype('D') p2 = PeriodDtype('D') assert is_dtype_equal(p1, p2) assert not is_dtype_equal(np.int64, p1) p3 = PeriodDtype('2D') assert not is_dtype_equal(p1, p3) assert not DatetimeTZDtype.is_dtype(np.int64) assert not PeriodDtype.is_dtype(np.int64)
def test_is_dtype(self): self.assertTrue(PeriodDtype.is_dtype(self.dtype)) self.assertTrue(PeriodDtype.is_dtype('period[D]')) self.assertTrue(PeriodDtype.is_dtype('period[3D]')) self.assertTrue(PeriodDtype.is_dtype(PeriodDtype('3D'))) self.assertTrue(PeriodDtype.is_dtype('period[U]')) self.assertTrue(PeriodDtype.is_dtype('period[S]')) self.assertTrue(PeriodDtype.is_dtype(PeriodDtype('U'))) self.assertTrue(PeriodDtype.is_dtype(PeriodDtype('S'))) self.assertFalse(PeriodDtype.is_dtype('D')) self.assertFalse(PeriodDtype.is_dtype('3D')) self.assertFalse(PeriodDtype.is_dtype('U')) self.assertFalse(PeriodDtype.is_dtype('S')) self.assertFalse(PeriodDtype.is_dtype('foo')) self.assertFalse(PeriodDtype.is_dtype(np.object_)) self.assertFalse(PeriodDtype.is_dtype(np.int64)) self.assertFalse(PeriodDtype.is_dtype(np.float64))