def testLabelMap(self):
with self.test_session() as sess:
default_1 = test_kernel_label_op.kernel_label()
# pylint: disable=protected-access
with sess.graph._kernel_label_map({"KernelLabel": "overload_1"}):
overload_1_1 = test_kernel_label_op.kernel_label()
with sess.graph._kernel_label_map({"KernelLabel": "overload_2"}):
overload_2 = test_kernel_label_op.kernel_label()
with sess.graph._kernel_label_map({"KernelLabel": ""}):
default_2 = test_kernel_label_op.kernel_label()
overload_1_2 = test_kernel_label_op.kernel_label()
# pylint: enable=protected-access
default_3 = test_kernel_label_op.kernel_label()
self.assertAllEqual("My label is: default", default_1.eval())
self.assertAllEqual("My label is: default", default_2.eval())
self.assertAllEqual("My label is: default", default_3.eval())
self.assertAllEqual("My label is: overload_1", overload_1_1.eval())
self.assertAllEqual("My label is: overload_1", overload_1_2.eval())
self.assertAllEqual("My label is: overload_2", overload_2.eval())
def testLabelMap(self):
with self.test_session() as sess:
default_1 = test_kernel_label_op.kernel_label()
# pylint: disable=protected-access
with sess.graph._kernel_label_map({"KernelLabel": "overload_1"}):
overload_1_1 = test_kernel_label_op.kernel_label()
with sess.graph._kernel_label_map(
{"KernelLabel": "overload_2"}):
overload_2 = test_kernel_label_op.kernel_label()
with sess.graph._kernel_label_map({"KernelLabel": ""}):
default_2 = test_kernel_label_op.kernel_label()
overload_1_2 = test_kernel_label_op.kernel_label()
# pylint: enable=protected-access
default_3 = test_kernel_label_op.kernel_label()
self.assertAllEqual("My label is: default", default_1.eval())
self.assertAllEqual("My label is: default", default_2.eval())
self.assertAllEqual("My label is: default", default_3.eval())
self.assertAllEqual("My label is: overload_1", overload_1_1.eval())
self.assertAllEqual("My label is: overload_1", overload_1_2.eval())
self.assertAllEqual("My label is: overload_2", overload_2.eval())
def testNoLabel(self):
with self.test_session():
self.assertAllEqual("My label is: default", test_kernel_label_op.kernel_label().eval())
def testNoLabel(self):
with self.test_session():
self.assertAllEqual(b"My label is: default",
test_kernel_label_op.kernel_label().eval())
