提交 93dee739 编写于 作者: T TensorFlower Gardener

Merge pull request #61842 from redwrasse:redwrasse/test-name-syntax-misc

PiperOrigin-RevId: 564914384
......@@ -40,7 +40,7 @@ class ZeroOut1Test(tf.test.TestCase):
result = zero_out_op_1.namespace_nested_zero_out([5, 4, 3, 2, 1])
self.assertAllEqual(result, [5, 0, 0, 0, 0])
def testLoadTwice(self):
def test_load_twice(self):
zero_out_loaded_again = tf.load_op_library(os.path.join(
tf.compat.v1.resource_loader.get_data_files_path(),
'zero_out_op_kernel_1.so'))
......
......@@ -25,15 +25,15 @@ class ZeroOut3Test(tf.test.TestCase):
result = zero_out_op_3.zero_out([5, 4, 3, 2, 1])
self.assertAllEqual(result, [5, 0, 0, 0, 0])
def testAttr(self):
def test_attr(self):
result = zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=3)
self.assertAllEqual(result, [0, 0, 0, 2, 0])
def testNegative(self):
def test_negative(self):
with self.assertRaisesOpError("Need preserve_index >= 0, got -1"):
self.evaluate(zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=-1))
def testLarge(self):
def test_large(self):
with self.assertRaisesOpError("preserve_index out of range"):
self.evaluate(zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=17))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册