def divtest():
x = Integer("Int1")
y = Integer("Int2")
z = x / y
graph = GradGraph(z)
output = graph.getOutput({x: 6,
y: 2})
return output == 3
def gradTestSimple():
a = Integer("a")
b = Integer("b")
e = (a + b) * (b + 1)
graph = GradGraph(e)
graph.getOutput({a: 2,
b: 1})
graph.getGradients(wrt=b)
return b.gradient == 5
def gradTestLong():
x = Integer("Int1x")
y = Integer("Int2y")
z = Integer("Int3z")
p = Integer("Int4p")
k = p * z
n = (k + (y * p * z)) * z
graph = GradGraph(n)
graph.getOutput({x: 9,
y: 9,
z: 9,
p: 2})
graph.getGradients(wrt=z)
return True
def gradTestShort():
x = Integer("Int1x")
y = Integer("Int2y")
z = Integer("Int3z")
p = Integer("Int4p")
k = p * z
t = y * k
m = k + t
n = m * z
graph = GradGraph(n)
graph.getOutput({x: 9,
y: 9,
z: 9,
p: 2})
graph.getGradients(wrt=z)
return z.gradient == 360
def test1():
x = Integer("Int1")
y = Integer("Int2")
z = Integer("Int3")
p = Integer("Int4")
k = p + z
kd = k * 2
t = x - kd
td = t - 2
s = td * z
sd = s / 5
graph = GradGraph(sd)
output = graph.getOutput({x: 36,
y: 2,
z: 3,
p: 9})
return output == 6
def testOps():
x = Integer('x')
y = ops.log(x)
z = ops.exp(y)
graph = GradGraph(z)
graph.getOutput({x: 1})
graph.getGradients(wrt=x)
return x.gradient == 1
def simpDiv():
x = Integer("Int1")
y = x / 4
graph = GradGraph(y)
output = graph.getOutput({x: 8})
return output == 2
def simpMul():
x = Integer("Int1")
y = x * 3
graph = GradGraph(y)
output = graph.getOutput({x: 2})
return output == 6
def simpSub():
x = Integer("Int1")
y = x - 3
graph = GradGraph(y)
output = graph.getOutput({x: 1})
return output == -2