def _only_type(cls, item: type) -> None:
# E.g. NDArray[int]
# The given item is the type of the single dimension.
from nptyping import get_type # Put here to prevent cyclic import.
cls._type = Any if item is Any else get_type(item)
def test_get_type_timedelta(self):
self.assertEqual(Timedelta64, get_type(timedelta(days=42)))
self.assertEqual(Timedelta64, get_type(timedelta))
self.assertEqual(Timedelta64, get_type(np.timedelta64))
self.assertEqual(Timedelta64, get_type(np.timedelta64()))
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_complex(self):
self.assertEqual(Complex128, get_type(1 + 1j))
self.assertEqual(Complex128, get_type(np.complex128(1 + 1j)))
self.assertEqual(Complex128, get_type(complex))
self.assertEqual(Complex128, get_type(np.complex128))
def test_get_type_datatime(self):
self.assertEqual(Datetime64, get_type(datetime.now()))
self.assertEqual(Datetime64, get_type(datetime))
self.assertEqual(Datetime64, get_type(np.datetime64))
self.assertEqual(Datetime64, get_type(np.datetime64()))
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_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_get_type_str(self):
self.assertEqual(Unicode[4], get_type('Test'))
def test_get_type_bool(self):
self.assertEqual(Bool, get_type(True))
def test_get_type_int(self):
self.assertEqual(Int[DEFAULT_INT_BITS], get_type(42))
def test_get_type_float(self):
self.assertEqual(Float[DEFAULT_FLOAT_BITS], get_type(42.0))
def test_get_type_some_random_class(self):
class SomeRandomClass:
...
self.assertEqual(Object, get_type(SomeRandomClass()))
def test_get_type_array(self):
self.assertEqual(NDArray[3, Int[DEFAULT_INT_BITS]],
get_type(np.array([1, 2, 3])))
def test_get_type_object(self):
self.assertEqual(Object, get_type(np.object))
def test_get_type_subarray_type(self):
self.assertEqual(SubArrayType[Int[32], (3,)], get_type(np.dtype((np.int32, 3))))
def test_get_type_structured_type(self):
self.assertEqual(StructuredType[Int[32]], get_type(np.dtype([('x', np.int32)])))
