def testMethods(self):
print('Methods testing')
print(np.arange(6, dtype=np.int32))
print(np.arange(6, dtype=np.float))
a = np.array([4, 3.])
print(type(a))
print(np.sort(a, None))
self.checkitems([3., 4], np.sort(a, None))
print(a.sort())
self.checkitems([3., 4], a)
a = np.array([4, 3.])
self.checkitems([1, 0], a.argsort())
self.checkitems([1, 0], np.argsort(a, None))
a = np.arange(6, dtype=np.float)
print(a.take([0, 2, 4]))
print(a.take([0, 2, 4], 0))
d = np.take(a, [0, 2, 4], 0)
print(type(d), d)
d = np.diag([0, 2, 3])
print(type(d), d)
a.shape = 2, 3
self.checkitems([1, 2], a.take([1, 2]))
self.checkitems([[1, 2], [4, 5]], a.take([1, 2], 1))
self.checkitems([0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5], a.repeat(2))
self.checkitems([[0, 1, 2], [0, 1, 2], [3, 4, 5], [3, 4, 5]],
a.repeat(2, axis=0))
self.checkitems([[0, 0, 1, 1, 2, 2], [3, 3, 4, 4, 5, 5]],
a.repeat(2, axis=1))
def __init__(self, axis=None, angle=None, R=None):
if isinstance(R, dnp.ndarray):
if R.shape != (3, 3):
raise Exception('R must be a 3 by 3 matrix')
self.R = R
angle = dnp.arccos(0.5*(sum(dnp.diag(R))-1))
self.angle = float(dnp.rad2deg(angle))
x = (R[2][1]-R[1][2])/(2*dnp.sin(angle))
y = (R[0][2]-R[2][0])/(2*dnp.sin(angle))
z = (R[1][0]-R[0][1])/(2*dnp.sin(angle))
axis = Vector(x,y,z)
axis = axis.unit()
self.axis = axis
return
if isinstance(angle, dnp.ndarray):
angle = angle[0]
elif isinstance(angle, int):
angle = float(angle)
if not isinstance(angle, float):
raise Exception('angle must be a float not {}'.format(type(angle)))
axis = axis.unit()
self.axis = axis
self.angle = angle
angle = float(dnp.radians(angle))
M = dnp.array([[0.0, -axis.z, axis.y],
[axis.z, 0.0, -axis.x],
[-axis.y, axis.x, 0.0]])
N = axis.tensor_product(axis)
I = dnp.identity(3)
self.R = I*dnp.cos(angle) + M*dnp.sin(angle) + (1-dnp.cos(angle))*N
angle = 56
def testMethods(self):
print('Methods testing')
print(np.arange(6, dtype=np.int32))
print(np.arange(6, dtype=np.float))
a = np.array([4,3.])
print(type(a))
print(np.sort(a, None))
self.checkitems([3.,4], np.sort(a, None))
print(a.sort())
self.checkitems([3.,4], a)
a = np.array([4,3.])
self.checkitems([1,0], a.argsort())
self.checkitems([1,0], np.argsort(a, None))
a = np.arange(6, dtype=np.float)
print(a.take([0, 2, 4]))
print(a.take([0, 2, 4], 0))
d = np.take(a, [0, 2, 4], 0)
print(type(d), d)
d = np.diag([0, 2, 3])
print(type(d), d)
a.shape = 2,3
self.checkitems([1,2], a.take([1,2]))
self.checkitems([[1,2], [4,5]], a.take([1,2],1))
self.checkitems([0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5], a.repeat(2))
self.checkitems([[0, 1, 2], [0, 1, 2], [3, 4, 5], [3, 4, 5]], a.repeat(2, axis=0))
self.checkitems([[0, 0, 1, 1, 2, 2], [3, 3, 4, 4, 5, 5]], a.repeat(2, axis=1))
def testMethods(self):
print np.arange(6, dtype=np.int32)
print np.arange(6, dtype=np.float)
a = np.array([4,3.])
print type(a)
print np.sort(a, None)
print a.sort()
a = np.arange(6, dtype=np.float)
print a.take([0, 2, 4], 0)
d = np.take(a, [0, 2, 4], 0)
print type(d), d
d = np.diag([0, 2, 3])
print type(d), d
def testMethods(self):
print 'Methods testing'
print np.arange(6, dtype=np.int32)
print np.arange(6, dtype=np.float)
a = np.array([4,3.])
print type(a)
print np.sort(a, None)
print a.sort()
a = np.arange(6, dtype=np.float)
print a.take([0, 2, 4])
print a.take([0, 2, 4], 0)
d = np.take(a, [0, 2, 4], 0)
print type(d), d
d = np.diag([0, 2, 3])
print type(d), d
a.shape = 2,3
self.checkitems([1,2], a.take([1,2]))
self.checkitems([[1,2], [4,5]], a.take([1,2],1))
self.checkitems([0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5], a.repeat(2))
self.checkitems([[0, 1, 2], [0, 1, 2], [3, 4, 5], [3, 4, 5]], a.repeat(2, axis=0))
self.checkitems([[0, 0, 1, 1, 2, 2], [3, 3, 4, 4, 5, 5]], a.repeat(2, axis=1))
def testMethods(self):
print 'Methods testing'
print np.arange(6, dtype=np.int32)
print np.arange(6, dtype=np.float)
a = np.array([4, 3.])
print type(a)
print np.sort(a, None)
print a.sort()
a = np.arange(6, dtype=np.float)
print a.take([0, 2, 4])
print a.take([0, 2, 4], 0)
d = np.take(a, [0, 2, 4], 0)
print type(d), d
d = np.diag([0, 2, 3])
print type(d), d
a.shape = 2, 3
self.checkitems([1, 2], a.take([1, 2]))
self.checkitems([[1, 2], [4, 5]], a.take([1, 2], 1))
self.checkitems([0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5], a.repeat(2))
self.checkitems([[0, 1, 2], [0, 1, 2], [3, 4, 5], [3, 4, 5]],
a.repeat(2, axis=0))
self.checkitems([[0, 0, 1, 1, 2, 2], [3, 3, 4, 4, 5, 5]],
a.repeat(2, axis=1))
