Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_get_type_numpy_dtype(self):
self.assertEqual(Int8, get_type(np.int8(42)))
self.assertEqual(Int16, get_type(np.int16(42)))
self.assertEqual(Int32, get_type(np.int32(42)))
self.assertEqual(Int64, get_type(np.int64(42)))
self.assertEqual(UInt8, get_type(np.uint8(42)))
self.assertEqual(UInt16, get_type(np.uint16(42)))
self.assertEqual(UInt32, get_type(np.uint32(42)))
self.assertEqual(UInt64, get_type(np.uint64(42)))
self.assertEqual(Float16, get_type(np.float16(42.0)))
self.assertEqual(Float32, get_type(np.float32(42.0)))
self.assertEqual(Float64, get_type(np.float64(42.0)))
self.assertEqual(Unicode, get_type(np.unicode))
self.assertEqual(Unicode[40], get_type(np.dtype(('U', 40))))
self.assertEqual(Bool, get_type(np.bool_(True)))
def test_get_type_numpy_type(self):
self.assertEqual(Int8, get_type(np.int8))
self.assertEqual(Int16, get_type(np.int16))
self.assertEqual(Int32, get_type(np.int32))
self.assertEqual(Int64, get_type(np.int64))
self.assertEqual(UInt8, get_type(np.uint8))
self.assertEqual(UInt16, get_type(np.uint16))
self.assertEqual(UInt32, get_type(np.uint32))
self.assertEqual(UInt64, get_type(np.uint64))
self.assertEqual(Float16, get_type(np.float16))
self.assertEqual(Float32, get_type(np.float32))
self.assertEqual(Float64, get_type(np.float64))
self.assertEqual(Bool, get_type(np.bool_))
def test_bits(self):
self.assertEqual(8, Int8.bits())
self.assertEqual(16, Int16.bits())
self.assertEqual(32, Int32.bits())
self.assertEqual(64, Int64.bits())
self.assertEqual(16, Float16.bits())
self.assertEqual(32, Float32.bits())
self.assertEqual(64, Float64.bits())
def test_int_fitting(self):
self.assertEqual(Int8, Int.fitting(0))
self.assertEqual(Int8, Int.fitting(2 ** 7 - 1))
self.assertEqual(Int8, Int.fitting(-(2 ** 7 - 1)))
self.assertEqual(Int16, Int.fitting(2 ** 7))
self.assertEqual(Int16, Int.fitting(2 ** 15 - 1))
self.assertEqual(Int16, Int.fitting(-(2 ** 7)))
self.assertEqual(Int16, Int.fitting(-(2 ** 15 - 1)))
self.assertEqual(Int32, Int.fitting(2 ** 15))
self.assertEqual(Int32, Int.fitting(2 ** 31 - 1))
self.assertEqual(Int32, Int.fitting(-(2 ** 15)))
self.assertEqual(Int32, Int.fitting(-(2 ** 31 - 1)))
self.assertEqual(Int64, Int.fitting(2 ** 31))
self.assertEqual(Int64, Int.fitting(2 ** 63 - 1))
self.assertEqual(Int64, Int.fitting(-(2 ** 31)))
self.assertEqual(Int64, Int.fitting(-(2 ** 63 - 1)))