source: branches/decorate/test_wrapper.py @ 855

#!/usr/bin/env python
#
# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
# Copyright (c) 2012-2016 California Institute of Technology.
# License: 3-clause BSD.  The full license text is available at:
#  - http://mmckerns.github.io/project/mystic/browser/mystic/LICENSE

from wrapper import *
from mystic.math import almostEqual

def test_monitored():

  from mystic.monitors import Monitor
  mon = Monitor()
  @monitored(monitor=mon)
  def squared(x):
    return x**2

  x = range(10)
  y = [squared(i) for i in x]
  assert y == mon.y
  assert x == mon.x


def test_counted():

  @counted
  def squared(x, n=0.0):
    return x**2 + n

  _ = squared(-1)
  _ = [squared(i) for i in range(10)]
  n,y = squared(10, 100)
  assert n[0] == 12
  assert y == 10**2 + 100

  from numpy import array
  x = array([1,2,3,4,5])
  n,y = squared(x, 1)
  assert n[0] == 13
  assert all(y == x**2 + 1)


def test_monitored_counted():

  from mystic.monitors import Monitor
  mon = Monitor()
  @monitoredcounted(monitor=mon)
  def squared(x):
    return x**2

  _ = squared(-1)
  _ = [squared(i) for i in range(10)]
  n,y = squared(10)
  assert n[0] == 12
  assert y == 10**2

  x = [-1] + range(10) + [10]
  assert mon.x == x
  assert mon.y == [i**2 for i in x]


def test_bounded():

  @bounded(min=1,max=9.9999)
  def squared(x):
    return x**2

  from numpy import array, inf
  x = range(11)
  y = [squared(i) for i in x]
  assert y == [i**2 if i in range(1,10) else inf for i in x]

  x = array(x)
  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 almostEqual(mean(set_mean(x, 5.0)), 5.0)


def test_mixedin():

  def squared(x, *args): #XXX: needs to have same number of arguments
    return x**2

  @mixedin(mixin=squared, shift=1, normalized=True)
  def plus_and_squared(x, n=0):
    return x + n

  assert plus_and_squared(2) == ((2**2) + (2+0))/2.0
  assert plus_and_squared(3,1) == ((3**2) + (3+1))/2.0


if __name__ == '__main__':
  test_monitored()
  test_counted()
  test_monitored_counted()
  test_bounded()
  test_mixedin()
  test_clip_bounded()
  test_target_bounded()
  test_bounce_bounded()


# EOF