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_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_eq(self):
self.assertTrue(Int8 == Int8)
self.assertTrue(Int16 == Int16)
self.assertTrue(Int32 == Int32)
self.assertTrue(Int64 == Int64)
self.assertTrue(UInt8 == UInt8)
self.assertTrue(UInt16 == UInt16)
self.assertTrue(UInt32 == UInt32)
self.assertTrue(UInt64 == UInt64)
self.assertTrue(Float16 == Float16)
self.assertTrue(Float32 == Float32)
self.assertTrue(Float64 == Float64)
self.assertTrue(Int8 != Int16)
self.assertTrue(Int32 != Float32)
self.assertTrue(Int32 != UInt32)
def test_uint_fitting(self):
self.assertEqual(UInt8, UInt.fitting(0))
self.assertEqual(UInt8, UInt.fitting(2 ** 8 - 1))
self.assertEqual(UInt8, UInt.fitting(-(2 ** 8 - 1)))
self.assertEqual(UInt16, UInt.fitting(2 ** 8))
self.assertEqual(UInt16, UInt.fitting(2 ** 16 - 1))
self.assertEqual(UInt16, UInt.fitting(-(2 ** 8)))
self.assertEqual(UInt16, UInt.fitting(-(2 ** 16 - 1)))
self.assertEqual(UInt32, UInt.fitting(2 ** 16))
self.assertEqual(UInt32, UInt.fitting(2 ** 32 - 1))
self.assertEqual(UInt32, UInt.fitting(-(2 ** 16)))
self.assertEqual(UInt32, UInt.fitting(-(2 ** 32 - 1)))
self.assertEqual(UInt64, UInt.fitting(2 ** 32))
self.assertEqual(UInt64, UInt.fitting(2 ** 64 - 1))
self.assertEqual(UInt64, UInt.fitting(-(2 ** 32)))
self.assertEqual(UInt64, UInt.fitting(-(2 ** 64 - 1)))
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)))
self.assertEqual(Bool, get_type(np.bool_(False)))