eja: add random_elements() method to get multiple random elements at once.

[sage.d.git] / mjo / eja / eja_algebra.py

diff --git a/mjo/eja/eja_algebra.py b/mjo/eja/eja_algebra.py
index 29ff1da38480b3b9b07d1ff39555948e31785732..1ae3286cea5a26eda3d82f7161aab639bc9d724c 100644 (file)
--- a/mjo/eja/eja_algebra.py
+++ b/mjo/eja/eja_algebra.py
@@ -5,6 +5,8 @@ are used in optimization, and have some additional nice methods beyond
 what can be supported in a general Jordan Algebra.
 """
 
+from itertools import repeat
+
 from sage.algebras.quatalg.quaternion_algebra import QuaternionAlgebra
 from sage.categories.magmatic_algebras import MagmaticAlgebras
 from sage.combinat.free_module import CombinatorialFreeModule
@@ -51,8 +53,7 @@ class FiniteDimensionalEuclideanJordanAlgebra(CombinatorialFreeModule):
 
             sage: set_random_seed()
             sage: J = random_eja()
-            sage: x = J.random_element()
-            sage: y = J.random_element()
+            sage: x,y = J.random_elements(2)
             sage: x*y == y*x
             True
 
@@ -446,9 +447,7 @@ class FiniteDimensionalEuclideanJordanAlgebra(CombinatorialFreeModule):
 
             sage: set_random_seed()
             sage: J = random_eja()
-            sage: x = J.random_element()
-            sage: y = J.random_element()
-            sage: z = J.random_element()
+            sage: x,y,z = J.random_elements(3)
             sage: (x*y).inner_product(z) == y.inner_product(x*z)
             True
 
@@ -657,6 +656,25 @@ class FiniteDimensionalEuclideanJordanAlgebra(CombinatorialFreeModule):
             s = super(FiniteDimensionalEuclideanJordanAlgebra, self)
             return s.random_element()
 
+    def random_elements(self, count):
+        """
+        Return ``count`` random elements as a tuple.
+
+        SETUP::
+
+            sage: from mjo.eja.eja_algebra import JordanSpinEJA
+
+        EXAMPLES::
+
+            sage: J = JordanSpinEJA(3)
+            sage: x,y,z = J.random_elements(3)
+            sage: all( [ x in J, y in J, z in J ])
+            True
+            sage: len( J.random_elements(10) ) == 10
+            True
+
+        """
+        return  tuple( self.random_element() for idx in xrange(count) )
 
     @classmethod
     def random_instance(cls, field=QQ, **kwargs):
@@ -824,8 +842,7 @@ class RealCartesianProductEJA(FiniteDimensionalEuclideanJordanAlgebra):
 
             sage: set_random_seed()
             sage: J = RealCartesianProductEJA.random_instance()
-            sage: x = J.random_element()
-            sage: y = J.random_element()
+            sage: x,y = J.random_elements(2)
             sage: X = x.natural_representation()
             sage: Y = y.natural_representation()
             sage: x.inner_product(y) == J.natural_inner_product(X,Y)
@@ -1032,8 +1049,7 @@ class RealSymmetricEJA(MatrixEuclideanJordanAlgebra):
 
         sage: set_random_seed()
         sage: J = RealSymmetricEJA.random_instance()
-        sage: x = J.random_element()
-        sage: y = J.random_element()
+        sage: x,y = J.random_elements(2)
         sage: actual = (x*y).natural_representation()
         sage: X = x.natural_representation()
         sage: Y = y.natural_representation()
@@ -1275,8 +1291,7 @@ class ComplexHermitianEJA(ComplexMatrixEuclideanJordanAlgebra):
 
         sage: set_random_seed()
         sage: J = ComplexHermitianEJA.random_instance()
-        sage: x = J.random_element()
-        sage: y = J.random_element()
+        sage: x,y = J.random_elements(2)
         sage: actual = (x*y).natural_representation()
         sage: X = x.natural_representation()
         sage: Y = y.natural_representation()
@@ -1525,8 +1540,7 @@ class QuaternionHermitianEJA(QuaternionMatrixEuclideanJordanAlgebra):
 
         sage: set_random_seed()
         sage: J = QuaternionHermitianEJA.random_instance()
-        sage: x = J.random_element()
-        sage: y = J.random_element()
+        sage: x,y = J.random_elements(2)
         sage: actual = (x*y).natural_representation()
         sage: X = x.natural_representation()
         sage: Y = y.natural_representation()
@@ -1691,8 +1705,7 @@ class JordanSpinEJA(FiniteDimensionalEuclideanJordanAlgebra):
 
             sage: set_random_seed()
             sage: J = JordanSpinEJA.random_instance()
-            sage: x = J.random_element()
-            sage: y = J.random_element()
+            sage: x,y = J.random_elements(2)
             sage: X = x.natural_representation()
             sage: Y = y.natural_representation()
             sage: x.inner_product(y) == J.natural_inner_product(X,Y)