Changeset 600 for branches

Timestamp:

11/25/12 05:10:28 (3 years ago)

Author:

mmckerns

Message:

add optimization test for penalty and (nested) constraints decorators

Location:

branches/decorate

Files:

• test_penalty.py (added)
• test_wrapper.py (modified) (4 diffs)

branches/decorate/test_wrapper.py

@@ -1 +1 @@
 from wrapper import *
+from mystic.math import almostEqual
 
 def test_nested():
@@ -155 +156 @@
   from numpy import array, arange
   x = range(11)
-  assert round(mean(set_mean(x, 5.0)), 15) == 5.0
+  assert almostEqual(mean(set_mean(x, 5.0)), 5.0)
 
 
@@ -181 +182 @@
   discrete_squared.samples([1.0, 7.0])
   assert discrete_squared(5.6) == 7.0**2
+
+
+def test_constrain():
+
+  from mystic.math.measures import mean, spread
+  from mystic.math.measures import impose_mean, impose_spread
+  def mean_constraint(x):
+    return impose_mean(5.0, x)
+
+  def range_constraint(x):
+    return impose_spread(5.0, x)
+
+  @nested(inner=range_constraint)  #XXX: @with_range(5.0)
+  @nested(inner=mean_constraint)   #XXX: @with_mean(5.0)
+  def constraints(x):
+    return x
+
+  def cost(x):
+    return abs(sum(x) - 5.0)
+
+  from mystic.solvers import fmin_powell
+  from numpy import array
+  x = array([1,2,3,4,5])
+  y = fmin_powell(cost, x, constraints=constraints, disp=False)
+
+  assert mean(y) == 5.0
+  assert spread(y) == 5.0
+  assert almostEqual(cost(y), 4*(5.0))
 
 
@@ -197 +226 @@
   test_target_bounded()
   test_bounce_bounded()
+  test_constrain()