Clean up some imports and fix another test failure.

diff --git a/mjo/all.py b/mjo/all.py
index 580cca52bcea2d84db06c1ebef228074aecbc3ad..1a46f595739c3e115cb1f25018df286594363c45 100644 (file)
--- a/mjo/all.py
+++ b/mjo/all.py
@@ -3,15 +3,17 @@ Import all of the other code, so that the user doesn't have to do it
 in his script. Instead, he can just `from mjo.all import *`.
 """
 
-from cone.cone import *
-from cone.completely_positive import *
-from cone.doubly_nonnegative import *
-from cone.permutation_invariant import *
-from cone.rearrangement import *
-from cone.symmetric_psd import *
-from interpolation import *
-from misc import *
-from orthogonal_polynomials import *
-from plot import *
-from symbol_sequence import *
-from symbolic import *
+# Sage doesn't load ~/.sage/init.sage during testing (sage -t), so we
+# have to explicitly mangle our sitedir here so that "mjo.foo"
+# resolves.
+from os.path import abspath
+from site import addsitedir
+addsitedir(abspath('../'))
+
+from mjo.cone.all import *
+from mjo.interpolation import *
+from mjo.misc import *
+from mjo.orthogonal_polynomials import *
+from mjo.plot import *
+from mjo.symbol_sequence import *
+from mjo.symbolic import *

diff --git a/mjo/cone/all.py b/mjo/cone/all.py
new file mode 100644 (file)
index 0000000..c8957a0
--- /dev/null
+++ b/mjo/cone/all.py
@@ -0,0 +1,16 @@
+"""
+All imports from mjo.cone modules.
+"""
+
+# Sage doesn't load ~/.sage/init.sage during testing (sage -t), so we
+# have to explicitly mangle our sitedir here so that "mjo.foo" resolves.
+from os.path import abspath
+from site import addsitedir
+addsitedir(abspath('../../'))
+
+from mjo.cone.cone import *
+from mjo.cone.completely_positive import *
+from mjo.cone.doubly_nonnegative import *
+from mjo.cone.permutation_invariant import *
+from mjo.cone.rearrangement import *
+from mjo.cone.symmetric_psd import *

diff --git a/mjo/cone/cone.py b/mjo/cone/cone.py
index 32e13861fdbb0cb2f806f5892cd9c90c1df051f4..49df3e9bc73b5071b4120f3ebfd41acb69921d01 100644 (file)
--- a/mjo/cone/cone.py
+++ b/mjo/cone/cone.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 is_lyapunov_like(L,K):

diff --git a/mjo/cone/symmetric_psd.py b/mjo/cone/symmetric_psd.py
index b3f05f62c707e5b6c7e02d8c2806e23f543f7d03..e4be629447987bca42a6d3e795f1b9926bc378c8 100644 (file)
--- a/mjo/cone/symmetric_psd.py
+++ b/mjo/cone/symmetric_psd.py
@@ -6,14 +6,6 @@ all symmetric positive-semidefinite matrices (as a subset of
 
 from sage.all import *
 
-# Sage doesn't load ~/.sage/init.sage during testing (sage -t), so we
-# have to explicitly mangle our sitedir here so that "mjo.symbolic"
-# resolves.
-from os.path import abspath
-from site import addsitedir
-addsitedir(abspath('../../'))
-
-
 def is_symmetric_psd(A):
     """
     Determine whether or not the matrix ``A`` is symmetric

diff --git a/mjo/symbolic.py b/mjo/symbolic.py
index e2645a30ed7fe33ceb8fb787d51bfaf87305f1ad..992260efb9aaedadf187e06f4db6941d5055cb7e 100644 (file)
--- a/mjo/symbolic.py
+++ b/mjo/symbolic.py
@@ -15,15 +15,15 @@ def set_simplification_domain(d):
 
     With the default 'complex' domain, we don't simplify this::
 
-        sage: (abs(x)^2).simplify()
-        abs(x)^2
+        sage: sqrt(x^2).simplify()
+        sqrt(x^2)
 
     But in the 'real' domain, we do::
 
         sage: set_simplification_domain('real')
         'real'
-        sage: (abs(x)^2).simplify()
-        x^2
+        sage: sqrt(x^2).simplify()
+        abs(x)
         sage: set_simplification_domain('complex')
         'complex'