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_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_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_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)))
def test_issubclass(self):
self.assertTrue(issubclass(Float[32], Float32))
self.assertTrue(issubclass(Int64, Int[64]))
self.assertTrue(issubclass(Float16, Float))
self.assertTrue(issubclass(Int64, Int))
self.assertTrue(issubclass(Int64, Number))
self.assertTrue(not issubclass(Int64, Float))
self.assertTrue(not issubclass(Float32, Float64))
self.assertTrue(issubclass(numpy.float64, Number))
self.assertTrue(issubclass(numpy.int32, Number))
self.assertTrue(issubclass(numpy.int32, Int32))
self.assertTrue(not issubclass(numpy.int32, Int64))
self.assertTrue(not issubclass(numpy.float32, Int32))
self.assertTrue(issubclass(int, Number))
self.assertTrue(issubclass(float, Number))
self.assertTrue(issubclass(int, Int[DEFAULT_INT_BITS]))