提交 0454f570 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Auto indentation.

上级 9f394ccf
...@@ -35,7 +35,7 @@ class MyOp(Op): ...@@ -35,7 +35,7 @@ class MyOp(Op):
self.name = name self.name = name
self.destroy_map = dmap self.destroy_map = dmap
self.x = x self.x = x
def make_node(self, *inputs): def make_node(self, *inputs):
inputs = map(as_variable, inputs) inputs = map(as_variable, inputs)
for input in inputs: for input in inputs:
...@@ -87,7 +87,7 @@ PatternOptimizer = lambda p1, p2, ign=False: OpKeyOptimizer(PatternSub(p1, p2), ...@@ -87,7 +87,7 @@ PatternOptimizer = lambda p1, p2, ign=False: OpKeyOptimizer(PatternSub(p1, p2),
TopoPatternOptimizer = lambda p1, p2, ign=True: TopoOptimizer(PatternSub(p1, p2), ignore_newtrees=ign) TopoPatternOptimizer = lambda p1, p2, ign=True: TopoOptimizer(PatternSub(p1, p2), ignore_newtrees=ign)
class TestPatternOptimizer: class TestPatternOptimizer:
def test_replace_output(self): def test_replace_output(self):
# replacing the whole graph # replacing the whole graph
x, y, z = inputs() x, y, z = inputs()
...@@ -96,7 +96,7 @@ class TestPatternOptimizer: ...@@ -96,7 +96,7 @@ class TestPatternOptimizer:
PatternOptimizer((op1, (op2, '1', '2'), '3'), PatternOptimizer((op1, (op2, '1', '2'), '3'),
(op4, '3', '2')).optimize(g) (op4, '3', '2')).optimize(g)
assert str(g) == "[Op4(z, y)]" assert str(g) == "[Op4(z, y)]"
def test_nested_out_pattern(self): def test_nested_out_pattern(self):
x, y, z = inputs() x, y, z = inputs()
e = op1(x, y) e = op1(x, y)
...@@ -239,7 +239,7 @@ class TestPatternOptimizer: ...@@ -239,7 +239,7 @@ class TestPatternOptimizer:
PatternOptimizer((op4, (op1, 'x', 'y')), PatternOptimizer((op4, (op1, 'x', 'y')),
(op3, 'x', 'y')).optimize(g) (op3, 'x', 'y')).optimize(g)
assert str(g) == "[Op3(Op4(*1 -> Op1(x, y)), *1)]" assert str(g) == "[Op3(Op4(*1 -> Op1(x, y)), *1)]"
def test_eq(self): def test_eq(self):
# replacing the whole graph # replacing the whole graph
x, y, z = inputs() x, y, z = inputs()
...@@ -265,14 +265,14 @@ OpSubOptimizer = lambda op1, op2: TopoOptimizer(OpSub(op1, op2)) ...@@ -265,14 +265,14 @@ OpSubOptimizer = lambda op1, op2: TopoOptimizer(OpSub(op1, op2))
OpSubOptimizer = lambda op1, op2: OpKeyOptimizer(OpSub(op1, op2)) OpSubOptimizer = lambda op1, op2: OpKeyOptimizer(OpSub(op1, op2))
class TestOpSubOptimizer: class TestOpSubOptimizer:
def test_straightforward(self): def test_straightforward(self):
x, y, z = inputs() x, y, z = inputs()
e = op1(op1(op1(op1(op1(x))))) e = op1(op1(op1(op1(op1(x)))))
g = Env([x, y, z], [e]) g = Env([x, y, z], [e])
OpSubOptimizer(op1, op2).optimize(g) OpSubOptimizer(op1, op2).optimize(g)
assert str(g) == "[Op2(Op2(Op2(Op2(Op2(x)))))]" assert str(g) == "[Op2(Op2(Op2(Op2(Op2(x)))))]"
def test_straightforward_2(self): def test_straightforward_2(self):
x, y, z = inputs() x, y, z = inputs()
e = op1(op2(x), op3(y), op4(z)) e = op1(op2(x), op3(y), op4(z))
...@@ -402,13 +402,3 @@ class TestEquilibrium(object): ...@@ -402,13 +402,3 @@ class TestEquilibrium(object):
_logger.setLevel(oldlevel) _logger.setLevel(oldlevel)
print 'after', g print 'after', g
assert str(g) == '[Op4(x, y)]' assert str(g) == '[Op4(x, y)]'
...@@ -4681,7 +4681,7 @@ class test_arithmetic_cast(unittest.TestCase): ...@@ -4681,7 +4681,7 @@ class test_arithmetic_cast(unittest.TestCase):
'default', 'default',
message='Division of two integer', message='Division of two integer',
category=DeprecationWarning) category=DeprecationWarning)
class test_broadcast(unittest.TestCase): class test_broadcast(unittest.TestCase):
def test_broadcast_bigdim(self): def test_broadcast_bigdim(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论