def test_compute_boundary_ts_1(self):
"""Test for ge.compute_boundary_ts."""
input_ts, output_ts, inside_ts = gde.compute_boundary_ts(
[self.g.op, self.h.op])
self.assertEqual(list(input_ts), [self.c, self.a, self.f])
self.assertEqual(list(output_ts), [self.h])
self.assertEqual(list(inside_ts), [self.g])
def test_compute_boundary_ts_2(self):
"""Test for ge.compute_boundary_ts."""
tf_graph = tf.Graph()
with tf_graph.as_default():
a_tensor = tf.constant(1, name="a")
b_tensor = tf.constant(1, name="b")
c_tensor = tf.add(a_tensor, b_tensor, name="c")
_ = a_tensor + c_tensor
g = gde.Graph(tf_graph)
input_ts, output_ts, inside_ts = gde.compute_boundary_ts([g["a"], g["c"]])
self.assertEqual(list(input_ts), [g["b"].output(0)])
self.assertEqual(list(output_ts), [g["a"].output(0), g["c"].output(0)])
self.assertEqual(list(inside_ts), [g["a"].output(0)])
def test_compute_boundary_ts_0(self): """Test for ge.compute_boundary_ts.""" input_ts, output_ts, inside_ts = gde.compute_boundary_ts(self.g.op) self.assertEqual(list(input_ts), [self.c, self.a]) self.assertEqual(list(output_ts), [self.g]) self.assertEqual(list(inside_ts), [])