def test_infer_dtype_from_scalar(self):
# Test that _infer_dtype_from_scalar is returning correct dtype for int
# and float.
for dtypec in [
np.uint8, np.int8, np.uint16, np.int16, np.uint32, np.int32,
np.uint64, np.int64
]:
data = dtypec(12)
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, type(data))
data = 12
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.int64)
for dtypec in [np.float16, np.float32, np.float64]:
data = dtypec(12)
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, dtypec)
data = np.float(12)
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.float64)
for data in [True, False]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.bool_)
for data in [np.complex64(1), np.complex128(1)]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.complex_)
import datetime
for data in [
np.datetime64(1, 'ns'),
Timestamp(1),
datetime.datetime(2000, 1, 1, 0, 0)
]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, 'M8[ns]')
for data in [
np.timedelta64(1, 'ns'),
Timedelta(1),
datetime.timedelta(1)
]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, 'm8[ns]')
for data in [
datetime.date(2000, 1, 1),
Timestamp(1, tz='US/Eastern'), 'foo'
]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.object_)
def test_infer_dtype_from_scalar(self):
# Test that _infer_dtype_from_scalar is returning correct dtype for int
# and float.
for dtypec in [np.uint8, np.int8, np.uint16, np.int16, np.uint32,
np.int32, np.uint64, np.int64]:
data = dtypec(12)
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, type(data))
data = 12
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.int64)
for dtypec in [np.float16, np.float32, np.float64]:
data = dtypec(12)
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, dtypec)
data = np.float(12)
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.float64)
for data in [True, False]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.bool_)
for data in [np.complex64(1), np.complex128(1)]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.complex_)
import datetime
for data in [np.datetime64(1, 'ns'), Timestamp(1),
datetime.datetime(2000, 1, 1, 0, 0)]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, 'M8[ns]')
for data in [np.timedelta64(1, 'ns'), Timedelta(1),
datetime.timedelta(1)]:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, 'm8[ns]')
for data in [datetime.date(2000, 1, 1),
Timestamp(1, tz='US/Eastern'), 'foo']:
dtype, val = _infer_dtype_from_scalar(data)
self.assertEqual(dtype, np.object_)