better use fixtures

python/tests/test_Rotation.py

@@ -1,54 +1,55 @@
+import pytest
 import numpy as np
+
 from damask import Rotation
+   
+n = 1000
+
+@pytest.fixture
+def default():
+    """A set of n random rotations."""
+    return [Rotation.fromRandom() for r in range(n)]
+
 
 class TestRotation:
-   
-    n = 1000
 
-    def test_Eulers(self):
+    def test_Eulers(self,default):
-        for r in range(self.n):
+        for rot in default:
-            rot = Rotation.fromRandom()
             assert np.allclose(rot.asQuaternion(), 
                                Rotation.fromEulers(rot.asEulers()).asQuaternion())
 
 
-    def test_AxisAngle(self):
+    def test_AxisAngle(self,default):
-        for r in range(self.n):
+        for rot in default:
-            rot = Rotation.fromRandom()
             assert np.allclose(rot.asEulers(),
                                Rotation.fromAxisAngle(rot.asAxisAngle()).asEulers())
 
 
-    def test_Matrix(self):
+    def test_Matrix(self,default):
-        for r in range(self.n):
+        for rot in default:
-            rot = Rotation.fromRandom()
             assert np.allclose(rot.asAxisAngle(),
                                Rotation.fromMatrix(rot.asMatrix()).asAxisAngle())
 
 
-    def test_Rodriques(self):
+    def test_Rodriques(self,default):
-        for r in range(self.n):
+        for rot in default:
-            rot = Rotation.fromRandom()
             assert np.allclose(rot.asMatrix(),
                                Rotation.fromRodrigues(rot.asRodrigues()).asMatrix())
 
 
-    def test_Homochoric(self):
+    def test_Homochoric(self,default):
-        for r in range(self.n):
+        for rot in default:
-            rot = Rotation.fromRandom()
             assert np.allclose(rot.asRodrigues(),
                                Rotation.fromHomochoric(rot.asHomochoric()).asRodrigues())
 
 
-    def test_Cubochoric(self):
+    def test_Cubochoric(self,default):
-        for r in range(self.n):
+        for rot in default:
-            rot = Rotation.fromRandom()
             assert np.allclose(rot.asHomochoric(),
                                Rotation.fromCubochoric(rot.asCubochoric()).asHomochoric())
 
 
-    def test_Quaternion(self):
+    def test_Quaternion(self,default):
-        for r in range(self.n):
+        for rot in default:
-            rot = Rotation.fromRandom()
             assert np.allclose(rot.asCubochoric(),
                                Rotation.fromQuaternion(rot.asQuaternion()).asCubochoric())