def test_translate_np_dtype(self):
'''
Tests dtypes.translate_np_dtype method
:return: None
:raise: AssertionError if 1.. test cases fail
'''
d_tuple = dtypes.translate_np_dtype(np.dtype(np.bool))
self.assertEqual(1, d_tuple[1])
self.assertEqual('bool', d_tuple[0])
d_tuple = dtypes.translate_np_dtype(np.dtype(np.int64))
self.assertEqual(8, d_tuple[1])
self.assertEqual('int', d_tuple[0])
d_tuple = dtypes.translate_np_dtype(np.dtype(np.float64))
self.assertEqual(8, d_tuple[1])
self.assertEqual('float', d_tuple[0])
d_tuple = dtypes.translate_np_dtype(np.dtype(np.uint8))
self.assertEqual(1, d_tuple[1])
self.assertEqual('uint', d_tuple[0])
d_tuple = dtypes.translate_np_dtype(np.dtype(np.str))
self.assertEqual(0, d_tuple[1])
self.assertEqual('str', d_tuple[0])
with self.assertRaises(TypeError):
dtypes.check_np_dtype(np.dtype(np.int16))
with self.assertRaises(TypeError):
dtypes.translate_np_dtype('np.str')
def test_check_np_dtype(self):
'''
Tests dtypes.check_np_dtype method
:return: None
:raise: AssertionError if 1.. test cases fail
'''
dtypes.check_np_dtype(np.dtype(np.bool))
dtypes.check_np_dtype(np.dtype(np.int64))
dtypes.check_np_dtype(np.dtype(np.float64))
dtypes.check_np_dtype(np.dtype(np.uint8))
dtypes.check_np_dtype(np.dtype(np.str))
with self.assertRaises(TypeError):
dtypes.check_np_dtype(np.dtype(np.int16))
with self.assertRaises(TypeError):
dtypes.check_np_dtype('np.str')
