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))