mjo/cone/permutation_invariant.py: fix tests with PYTHONPATH="."

diff --git a/mjo/cone/permutation_invariant.py b/mjo/cone/permutation_invariant.py
index ff04d7e3152b8c3a856c6310b67cfe25e0d967e9..d6642806c0e0d8473908c5f761a75556085e0653 100644 (file)
--- a/mjo/cone/permutation_invariant.py
+++ b/mjo/cone/permutation_invariant.py
@@ -60,9 +60,13 @@ def is_permutation_invariant(K):
     A cone is said to be permutation-invariant if ``P(K)`` is a subset
     of ``K`` for every permutation matrix ``P``.
 
-    EXAMPLES:
+    SETUP::
 
         sage: from mjo.cone.rearrangement import rearrangement_cone
+        sage: from mjo.cone.permutation_invariant import is_permutation_invariant
+
+    EXAMPLES::
+
         sage: all([ is_permutation_invariant(rearrangement_cone(p,n))
         ....:               for n in range(3, 6)
         ....:               for p in range(1, n) ])