def test_toArray_radii(self):
"""MageList toArray should return the correct array"""
m = MageList(self.empty_list)
self.assertEqual(m.toArray(),array(()))
m = MageList(self.first_list,Radius=.3)
self.assertEqual(m.toArray(),array([[0,0,0,0.3],[2.0,4.0,6.0,0.3]]))
m = MageList(self.radii)
self.assertEqual(m.toArray(), array([[2,2,2,.1],[3,3,3,.5]]))
m = MageList(self.radii,Radius=.4)
self.assertEqual(m.toArray(), array([[2,2,2,.1],[3,3,3,.5]]))
m = MageList(self.single_list) #radius = None
self.assertRaises(ValueError,m.toArray)
def test_toArray_coords_only(self):
"""MageList toArray should return the correct array"""
m = MageList(self.empty_list)
self.assertEqual(m.toArray(include_radius=False),array(()))
m = MageList(self.first_list,Radius=.3)
self.assertEqual(m.toArray(include_radius=False),
array([[0,0,0],[2.0,4.0,6.0]]))
m = MageList(self.radii)
self.assertEqual(m.toArray(include_radius=False),
array([[2,2,2],[3,3,3]]))
m = MageList(self.radii,Radius=.4)
self.assertEqual(m.toArray(include_radius=False),
array([[2,2,2],[3,3,3]]))
m = MageList(self.single_list) #radius = None
self.assertEqual(m.toArray(include_radius=False),array([[1,1,1]]))
