]> gitweb.michael.orlitzky.com - dunshire.git/blobdiff - src/dunshire/symmetric_linear_game.py
Add a first unit test to check that a solution is always obtained.
[dunshire.git] / src / dunshire / symmetric_linear_game.py
index 968d9ca630c0af2fa15e780747785e9705f82ddc..b6489ba6de43877dd5707a39b3d7811dd9d2e4a7 100644 (file)
@@ -6,13 +6,15 @@ to solve a linear game.
 """
 
 from cvxopt import matrix, printing, solvers
+from unittest import TestCase
 
 from cones import CartesianProduct
 from errors import GameUnsolvableException
-from matrices import append_col, append_row, identity
+from matrices import append_col, append_row, identity, inner_product
+import options
 
-printing.options['dformat'] = '%.7f'
-solvers.options['show_progress'] = False
+printing.options['dformat'] = options.FLOAT_FORMAT
+solvers.options['show_progress'] = options.VERBOSE
 
 
 class Solution:
@@ -106,22 +108,106 @@ class SymmetricLinearGame:
         """
         INPUT:
 
-          - ``L`` -- an n-by-b matrix represented as a list of lists
-             of real numbers.
+          - ``L`` -- an square matrix represented as a list of lists
+             of real numbers. ``L`` itself is interpreted as a list of
+             ROWS, which agrees with (for example) SageMath and NumPy,
+             but not with CVXOPT (whose matrix constructor accepts a
+             list of columns).
 
           - ``K`` -- a SymmetricCone instance.
 
-          - ``e1`` -- the interior point of ``K`` belonging to player one,
-                      as a column vector.
+          - ``e1`` -- the interior point of ``K`` belonging to player one;
+            it can be of any enumerable type having the correct length.
 
-          - ``e2`` -- the interior point of ``K`` belonging to player two,
-                      as a column vector.
+          - ``e2`` -- the interior point of ``K`` belonging to player two;
+            it can be of any enumerable type having the correct length.
 
+        EXAMPLES:
+
+        Lists can (and probably should) be used for every argument:
+
+            >>> from cones import NonnegativeOrthant
+            >>> K = NonnegativeOrthant(2)
+            >>> L = [[1,0],[0,1]]
+            >>> e1 = [1,1]
+            >>> e2 = [1,1]
+            >>> G = SymmetricLinearGame(L, K, e1, e2)
+            >>> print(G)
+            The linear game (L, K, e1, e2) where
+              L = [ 1  0]
+                  [ 0  1],
+              K = Nonnegative orthant in the real 2-space,
+              e1 = [ 1]
+                   [ 1],
+              e2 = [ 1]
+                   [ 1].
+
+        The points ``e1`` and ``e2`` can also be passed as some other
+        enumerable type (of the correct length) without much harm, since
+        there is no row/column ambiguity:
+
+            >>> import cvxopt
+            >>> import numpy
+            >>> from cones import NonnegativeOrthant
+            >>> K = NonnegativeOrthant(2)
+            >>> L = [[1,0],[0,1]]
+            >>> e1 = cvxopt.matrix([1,1])
+            >>> e2 = numpy.matrix([1,1])
+            >>> G = SymmetricLinearGame(L, K, e1, e2)
+            >>> print(G)
+            The linear game (L, K, e1, e2) where
+              L = [ 1  0]
+                  [ 0  1],
+              K = Nonnegative orthant in the real 2-space,
+              e1 = [ 1]
+                   [ 1],
+              e2 = [ 1]
+                   [ 1].
+
+        However, ``L`` will always be intepreted as a list of rows, even
+        if it is passed as a ``cvxopt.base.matrix`` which is otherwise
+        indexed by columns:
+
+            >>> import cvxopt
+            >>> from cones import NonnegativeOrthant
+            >>> K = NonnegativeOrthant(2)
+            >>> L = [[1,2],[3,4]]
+            >>> e1 = [1,1]
+            >>> e2 = e1
+            >>> G = SymmetricLinearGame(L, K, e1, e2)
+            >>> print(G)
+            The linear game (L, K, e1, e2) where
+              L = [ 1  2]
+                  [ 3  4],
+              K = Nonnegative orthant in the real 2-space,
+              e1 = [ 1]
+                   [ 1],
+              e2 = [ 1]
+                   [ 1].
+            >>> L = cvxopt.matrix(L)
+            >>> print(L)
+            [ 1  3]
+            [ 2  4]
+            <BLANKLINE>
+            >>> G = SymmetricLinearGame(L, K, e1, e2)
+            >>> print(G)
+            The linear game (L, K, e1, e2) where
+              L = [ 1  2]
+                  [ 3  4],
+              K = Nonnegative orthant in the real 2-space,
+              e1 = [ 1]
+                   [ 1],
+              e2 = [ 1]
+                   [ 1].
         """
         self._K = K
         self._e1 = matrix(e1, (K.dimension(), 1))
         self._e2 = matrix(e2, (K.dimension(), 1))
-        self._L = matrix(L, (K.dimension(), K.dimension()))
+
+        # Our input ``L`` is indexed by rows but CVXOPT matrices are
+        # indexed by columns, so we need to transpose the input before
+        # feeding it to CVXOPT.
+        self._L = matrix(L, (K.dimension(), K.dimension())).trans()
 
         if not K.contains_strict(self._e1):
             raise ValueError('the point e1 must lie in the interior of K')
@@ -137,7 +223,7 @@ class SymmetricLinearGame:
 
             >>> from cones import NonnegativeOrthant
             >>> K = NonnegativeOrthant(3)
-            >>> L = [[1,-1,-12],[-5,2,-15],[-15,-3,1]]
+            >>> L = [[1,-5,-15],[-1,2,-3],[-12,-15,1]]
             >>> e1 = [1,1,1]
             >>> e2 = [1,2,3]
             >>> SLG = SymmetricLinearGame(L, K, e1, e2)
@@ -160,10 +246,10 @@ class SymmetricLinearGame:
               '  K = {!s},\n' \
               '  e1 = {:s},\n' \
               '  e2 = {:s}.'
-        L_str = '\n      '.join(str(self._L).splitlines())
-        e1_str = '\n       '.join(str(self._e1).splitlines())
-        e2_str = '\n       '.join(str(self._e2).splitlines())
-        return tpl.format(L_str, str(self._K), e1_str, e2_str)
+        indented_L = '\n      '.join(str(self._L).splitlines())
+        indented_e1 = '\n       '.join(str(self._e1).splitlines())
+        indented_e2 = '\n       '.join(str(self._e2).splitlines())
+        return tpl.format(indented_L, str(self._K), indented_e1, indented_e2)
 
 
     def solution(self):
@@ -186,7 +272,7 @@ class SymmetricLinearGame:
 
             >>> from cones import NonnegativeOrthant
             >>> K = NonnegativeOrthant(3)
-            >>> L = [[1,-1,-12],[-5,2,-15],[-15,-3,1]]
+            >>> L = [[1,-5,-15],[-1,2,-3],[-12,-15,1]]
             >>> e1 = [1,1,1]
             >>> e2 = [1,1,1]
             >>> SLG = SymmetricLinearGame(L, K, e1, e2)
@@ -277,7 +363,7 @@ class SymmetricLinearGame:
 
             >>> from cones import NonnegativeOrthant
             >>> K = NonnegativeOrthant(3)
-            >>> L = [[1,-1,-12],[-5,2,-15],[-15,-3,1]]
+            >>> L = [[1,-5,-15],[-1,2,-3],[-12,-15,1]]
             >>> e1 = [1,1,1]
             >>> e2 = [1,2,3]
             >>> SLG = SymmetricLinearGame(L, K, e1, e2)
@@ -295,7 +381,39 @@ class SymmetricLinearGame:
                    [ 1].
 
         """
-        return SymmetricLinearGame(self._L.trans(),
+        return SymmetricLinearGame(self._L, # It will be transposed in __init__().
                                    self._K, # Since "K" is symmetric.
                                    self._e2,
                                    self._e1)
+
+
+class SymmetricLinearGameTest(TestCase):
+
+    def assertEqualWithinTol(self, first, second):
+        """
+        Test that ``first`` and ``second`` are equal within our default
+        tolerance.
+        """
+        self.assertTrue(abs(first - second) < options.ABS_TOL)
+
+
+    def test_solution_exists(self):
+        """
+        Every linear game has a solution, so we should be able to solve
+        every symmetric linear game. Pick some parameters randomly and
+        give it a shot.
+        """
+        from cones import NonnegativeOrthant
+        from random import randint, uniform
+        ambient_dim = randint(1,10)
+        K = NonnegativeOrthant(ambient_dim)
+        e1 = [uniform(0.1, 10) for idx in range(ambient_dim)]
+        e2 = [uniform(0.1, 10) for idx in range(ambient_dim)]
+        L = [[uniform(-10, 10) for i in range(ambient_dim)]
+              for j in range(ambient_dim)]
+        G = SymmetricLinearGame(L, K, e1, e2)
+        soln = G.solution()
+        L_matrix = matrix(L).trans()
+        expected = inner_product(L_matrix*soln.player1_optimal(),
+                                 soln.player2_optimal())
+        self.assertEqualWithinTol(soln.game_value(), expected)