X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=mjo%2Fcone%2Frearrangement.py;h=b32938117bbf176450d7aeefe246cbb5e6731361;hb=785d079a2f5faa5126685aea96d584c3fa9c3007;hp=9dde4ec2fce949bb4f8e99fb76ecf79352cf1611;hpb=3cfc8e228ae337aed975118444a8cbad9a5a7ac3;p=sage.d.git

diff --git a/mjo/cone/rearrangement.py b/mjo/cone/rearrangement.py
index 9dde4ec..b329381 100644
--- a/mjo/cone/rearrangement.py
+++ b/mjo/cone/rearrangement.py
@@ -1,12 +1,4 @@
-# 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 *
-from mjo.cone.cone import random_element
 
 def rearrangement_cone(p,n):
     r"""
@@ -36,6 +28,10 @@ def rearrangement_cone(p,n):
     A polyhedral closed convex cone object representing a rearrangement
     cone of order ``p`` in ``n`` dimensions.
 
+    SETUP::
+
+        sage: from mjo.cone.rearrangement import rearrangement_cone
+
     EXAMPLES:
 
     The rearrangement cones of order one are nonnegative orthants::
@@ -127,6 +123,11 @@ def has_rearrangement_property(v, p):
     components sum to a nonnegative number), ``True`` is returned. Otherwise
     ``False`` is returned.
 
+    SETUP::
+
+        sage: from mjo.cone.rearrangement import (has_rearrangement_property,
+        ....:                                     rearrangement_cone)
+
     EXAMPLES:
 
     Every element of a rearrangement cone should have the property::
@@ -134,8 +135,9 @@ def has_rearrangement_property(v, p):
         sage: for n in range(2,10):
         ....:     for p in range(1, n-1):
         ....:         K = rearrangement_cone(p,n)
-        ....:         v = random_element(K)
-        ....:         if not has_rearrangement_property(v,p): print v
+        ....:         v = K.random_element()
+        ....:         if not has_rearrangement_property(v,p):
+        ....:             print(v)
 
     """
     components = sorted(v)[0:p]