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