Changeset 598 for branches/decorate/test_wrapper.py

Timestamp:

11/21/12 19:22:46 (3 years ago)

Author:

mmckerns

Message:

added clip_bounded, target_bounded, and bounce_bounded decorators to wrapper

File:

• branches/decorate/test_wrapper.py (modified) (2 diffs)

branches/decorate/test_wrapper.py

@@ -113 +113 @@
   y = squared(x)
   assert y == inf #XXX: better, [i**2 if i in range(1,10) else inf for i in x] ?
+
+
+def test_clip_bounded():
+
+  @clip_bounded(min=1.0,max=9.9999)
+  def squared(x):
+    return x**2
+
+  from numpy import array, arange
+  x = range(11)
+  y = [squared(i) for i in x]
+  assert y == [min(max(i,1.0),9.9999)**2 for i in x]
+
+  y = squared(x)
+  assert all(y == array([min(max(i,1.0),9.9999)**2 for i in x]))
+
+
+def test_target_bounded():
+
+  @target_bounded(min=1.0,max=9.9999,target=5.0)
+  def squared(x):
+    return x**2
+
+  from numpy import array, arange
+  x = range(11)
+  y = [squared(i) for i in x]
+  assert y == [i**2 if 1.0 <= i <= 9.999 else 5.0**2 for i in x]
+
+  y = squared(x)
+  assert all(y == [i**2 if 1.0 <= i <= 9.9999 else 5.0**2 for i in x])
+
+
+def test_bounce_bounded():
+
+  from mystic.math.measures import impose_mean, mean
+
+  @bounce_bounded(min=1.0,max=9.9999)
+  def set_mean(x, m):
+    return impose_mean(m, x)
+
+  from numpy import array, arange
+  x = range(11)
+  assert round(mean(set_mean(x, 5.0)), 15) == 5.0
 
 
@@ -151 +194 @@
   test_mixedin()
   test_discrete()
+  test_clip_bounded()
+  test_target_bounded()
+  test_bounce_bounded()