def test_start_end_jobs():
x = theano.tensor.matrix('x')
y = theano.tensor.dot(x, x); y.name = 'y'
dag, dinputs, doutputs = dicdag.theano.theano_graph_to_dag((x,), (y,))
(dx,) = dinputs
(dy,) = doutputs
assert dx.name == x.name
assert dy.name == y.name
dag2 = merge(start_jobs(dinputs), end_jobs(doutputs), dag)
assert dy in dag2
assert any(len(v['args'])==1 and v['args'][0] == dy for v in dag2.values())
unidag = dicdag.unidag.dag_to_unidag(dag2)
def test_start_end_jobs():
x = theano.tensor.matrix('x')
y = theano.tensor.dot(x, x)
y.name = 'y'
dag, dinputs, doutputs = dicdag.theano.theano_graph_to_dag((x, ), (y, ))
(dx, ) = dinputs
(dy, ) = doutputs
assert dx.name == x.name
assert dy.name == y.name
dag2 = merge(start_jobs(dinputs), end_jobs(doutputs), dag)
assert dy in dag2
assert any(
len(v['args']) == 1 and v['args'][0] == dy for v in dag2.values())
unidag = dicdag.unidag.dag_to_unidag(dag2)
def test_start_jobs():
assert all(is_start_job(j)
for j in dicdag.unidag.dag_to_unidag(start_jobs('abcd')))
def test_start_jobs():
assert all(
is_start_job(j)
for j in dicdag.unidag.dag_to_unidag(start_jobs('abcd')))