cone/schur.py: use generator expressions where applicable.

[sage.d.git] / mjo / cone / permutation_invariant.py

diff --git a/mjo/cone/permutation_invariant.py b/mjo/cone/permutation_invariant.py
index ff04d7e3152b8c3a856c6310b67cfe25e0d967e9..7aad31e9e709114524bedd3f1354219867a99c57 100644 (file)
--- a/mjo/cone/permutation_invariant.py
+++ b/mjo/cone/permutation_invariant.py
@@ -1,10 +1,3 @@
-# Sage doesn't load ~/.sage/init.sage during testing (sage -t), so we
-# have to explicitly mangle our sitedir here so that "mjo.cone"
-# resolves.
-from os.path import abspath
-from site import addsitedir
-addsitedir(abspath('../../'))
-
 from sage.all import *
 
 def random_permutation_invariant_cone(lattice=None,
@@ -60,9 +53,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) ])