def test_foo():
x = np.zeros(3)
w = Watcher([x])
y = w.call(compute_stuff, x)
assert y == 12
f = w.recalculate_fn(y, x)
assert f(x) == 12
assert f(x + 1) != 12
def test_foo():
x = np.zeros(3)
w = Watcher([x])
y = w.call(compute_stuff, x)
assert y == 12
f = w.recalculate_fn(y, x)
assert f(x) == 12
assert f(x + 1) != 12
def test_grad():
x = np.zeros(3)
w = Watcher([x])
assert id(x) in w.svars
y = w.call(compute_stuff, x)
assert id(y) in w.svars
dy_dx_fn = w.grad_fn(y, x)
print dy_dx_fn(x)
print dy_dx_fn(x + 1)
print dy_dx_fn(x + 2)
def test_grad():
x = np.zeros(3)
w = Watcher([x])
assert id(x) in w.svars
y = w.call(compute_stuff, x)
assert id(y) in w.svars
dy_dx_fn = w.grad_fn(y, x)
print dy_dx_fn(x)
print dy_dx_fn(x + 1)
print dy_dx_fn(x + 2)
def test_loop():
def repeat_double(x, N):
print 'N', N
for i in range(N):
x = x + x
print 'i', i, 'x', x
assert i == N - 1
return x
#repeat_double(0, 4)
x = np.zeros(3)
w = Watcher([x])
y = w.call(repeat_double, x, 4)
f = w.recalculate_fn(y, x)
y2 = f(x + 1)
assert np.all(y == 0)
assert np.all(y2 == 16)
def test_loop():
def repeat_double(x, N):
print 'N', N
for i in range(N):
x = x + x
print 'i', i, 'x', x
assert i == N - 1
return x
#repeat_double(0, 4)
x = np.zeros(3)
w = Watcher([x])
y = w.call(repeat_double, x, 4)
f = w.recalculate_fn(y, x)
y2 = f(x + 1)
assert np.all(y == 0)
assert np.all(y2 == 16)
