polishing

This commit is contained in:
Martin Diehl 2020-06-30 23:17:50 +02:00
parent ef0c78745a
commit 23365660d8
3 changed files with 8 additions and 8 deletions

View File

@ -40,6 +40,7 @@ class Orientation: # ToDo: make subclass of lattice and Rotation
self.rotation = Rotation.from_quaternion(rotation) # assume quaternion self.rotation = Rotation.from_quaternion(rotation) # assume quaternion
def __getitem__(self,item): def __getitem__(self,item):
"""Iterate over leading/leftmost dimension of Orientation array."""
return self.__class__(self.rotation[item],self.lattice) return self.__class__(self.rotation[item],self.lattice)

View File

@ -81,17 +81,16 @@ class Rotation:
]) ])
def __len__(self):
return 0 if self.shape == () else len(self.shape)
def __getitem__(self,item): def __getitem__(self,item):
"""Iterate over leading/leftmost dimension of Rotation array."""
if self.shape == (): return self.copy()
if isinstance(item,tuple) and len(item) >= len(self): if isinstance(item,tuple) and len(item) >= len(self):
raise IndexError('Too many indices') raise IndexError('Too many indices')
return self.__class__(self.quaternion[item]) return self.__class__(self.quaternion[item])
def __len__(self): def __len__(self):
"""Length of leading/leftmost dimension of Rotation array."""
return 0 if self.shape == () else self.shape[0] return 0 if self.shape == () else self.shape[0]
@ -104,9 +103,10 @@ class Rotation:
other : numpy.ndarray or Rotation other : numpy.ndarray or Rotation
Vector, second or fourth order tensor, or rotation object that is rotated. Vector, second or fourth order tensor, or rotation object that is rotated.
Todo Returns
---- -------
Check rotation of 4th order tensor other_rot : numpy.ndarray or Rotation
Rotated vector, second or fourth order tensor, or rotation object.
""" """
if isinstance(other, Rotation): if isinstance(other, Rotation):

View File

@ -3,7 +3,6 @@ import random
import pytest import pytest
import numpy as np import numpy as np
from damask import Orientation
from damask import Rotation from damask import Rotation
from damask import Symmetry from damask import Symmetry