def test_nd_to_sa_no_type(self):
nd = np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]], dtype=float)
dtype = np.dtype({'names': map('f{}'.format, xrange(3)),
'formats': [float] * 3})
control = np.array([(1.0, 2.0, 3.0), (4.0, 5.0, 6.0)], dtype=dtype)
result = utils.cast_np_nd_to_sa(nd)
self.assertEqual(control.dtype, result.dtype)
self.assertTrue(np.array_equal(result, control))
def test_nd_to_sa_w_type(self):
nd = np.array([[1, 2, 3], [4, 5, 6]], dtype=int)
dtype = np.dtype({'names': map('f{}'.format, xrange(3)),
'formats': [int] * 3})
control = np.array([(1, 2, 3), (4, 5, 6)], dtype=dtype)
result = utils.cast_np_nd_to_sa(nd, dtype)
self.assertEqual(control.dtype, result.dtype)
self.assertTrue(np.array_equal(result, control))
def test_nd_to_sa_no_type(self):
nd = np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]], dtype=float)
dtype = np.dtype({
'names': map('f{}'.format, xrange(3)),
'formats': [float] * 3
})
control = np.array([(1.0, 2.0, 3.0), (4.0, 5.0, 6.0)], dtype=dtype)
result = utils.cast_np_nd_to_sa(nd)
self.assertEqual(control.dtype, result.dtype)
self.assertTrue(np.array_equal(result, control))
def test_nd_to_sa_w_type(self):
nd = np.array([[1, 2, 3], [4, 5, 6]], dtype=int)
dtype = np.dtype({
'names': map('f{}'.format, xrange(3)),
'formats': [int] * 3
})
control = np.array([(1, 2, 3), (4, 5, 6)], dtype=dtype)
result = utils.cast_np_nd_to_sa(nd, dtype)
self.assertEqual(control.dtype, result.dtype)
self.assertTrue(np.array_equal(result, control))