def test_pandas_dtype(): assert pandas_dtype('datetime64[ns, US/Eastern]') == DatetimeTZDtype( 'datetime64[ns, US/Eastern]') assert pandas_dtype('category') == CategoricalDtype() for dtype in ['M8[ns]', 'm8[ns]', 'object', 'float64', 'int64']: assert pandas_dtype(dtype) == np.dtype(dtype)
def test_hash_vs_equality(self): # make sure that we satisfy is semantics dtype = self.dtype dtype2 = CategoricalDtype() self.assertTrue(dtype == dtype2) self.assertTrue(dtype2 == dtype) self.assertTrue(dtype is dtype2) self.assertTrue(dtype2 is dtype) self.assertTrue(hash(dtype) == hash(dtype2))
def test_is_dtype(self): self.assertTrue(CategoricalDtype.is_dtype(self.dtype)) self.assertTrue(CategoricalDtype.is_dtype("category")) self.assertTrue(CategoricalDtype.is_dtype(CategoricalDtype())) self.assertFalse(CategoricalDtype.is_dtype("foo")) self.assertFalse(CategoricalDtype.is_dtype(np.float64))
def test_construction_from_string(self): result = CategoricalDtype.construct_from_string("category") self.assertTrue(is_dtype_equal(self.dtype, result)) self.assertRaises(TypeError, lambda: CategoricalDtype.construct_from_string("foo"))
def test_is_dtype(self): self.assertTrue(CategoricalDtype.is_dtype(self.dtype)) self.assertTrue(CategoricalDtype.is_dtype('category')) self.assertTrue(CategoricalDtype.is_dtype(CategoricalDtype())) self.assertFalse(CategoricalDtype.is_dtype('foo')) self.assertFalse(CategoricalDtype.is_dtype(np.float64))
def test_construction_from_string(self): result = CategoricalDtype.construct_from_string('category') self.assertTrue(is_dtype_equal(self.dtype, result)) self.assertRaises( TypeError, lambda: CategoricalDtype.construct_from_string('foo'))
def test_equality(self): self.assertTrue(is_dtype_equal(self.dtype, 'category')) self.assertTrue(is_dtype_equal(self.dtype, CategoricalDtype())) self.assertFalse(is_dtype_equal(self.dtype, 'foo'))
def setUp(self): self.dtype = CategoricalDtype()