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_isinstance(self):
self.assertIsInstance(42, Int)
self.assertIsInstance(42, Int32)
self.assertIsInstance(42.0, Float)
self.assertIsInstance(42.0, Float64)
self.assertIsInstance(42, Number)
self.assertIsInstance(Int64, Int[64])
self.assertIsInstance(Float64, Float[64])
self.assertIsInstance(Int64, Int)
self.assertIsInstance(Float64, Float)
self.assertNotIsInstance(Int, Float)
self.assertNotIsInstance(Float, Int)
self.assertIsInstance(numpy.int64(42), Int[64])
self.assertIsInstance(numpy.int64(42), Int64)
self.assertNotIsInstance(numpy.int32(42), Int64)
self.assertIsInstance(numpy.float64(42), Float[64])
self.assertIsInstance(numpy.float64(42), Float64)
self.assertNotIsInstance(numpy.float32(42), Float64)
def test_get_type_nptype(self):
self.assertEqual(Int32, get_type(Int32))
self.assertEqual(Float64, get_type(Float64))
self.assertEqual(Unicode[100], get_type(Unicode[100]))
self.assertEqual(Bool, get_type(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())