X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=mjo%2Fcone%2Fpermutation_invariant.py;h=7aad31e9e709114524bedd3f1354219867a99c57;hb=c41a2b88c599484041267165cf7f83fc880e4105;hp=ff04d7e3152b8c3a856c6310b67cfe25e0d967e9;hpb=5d4c579d47ef0c9228ae1d528850e92d77aaa9b0;p=sage.d.git

diff --git a/mjo/cone/permutation_invariant.py b/mjo/cone/permutation_invariant.py
index ff04d7e..7aad31e 100644
--- 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) ])