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()
