def testLarge(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))
def testAttr(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): 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): with self.test_session(): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=17) with self.assertRaisesOpError("preserve_index out of range"): result.eval()
def test(self): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1]) self.assertAllEqual(result, [5, 0, 0, 0, 0])
def testAttr(self): with self.test_session(): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=3) self.assertAllEqual(result.eval(), [0, 0, 0, 2, 0])
def testNegative(self): with self.test_session(): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=-1) with self.assertRaisesOpError("Need preserve_index >= 0, got -1"): result.eval()
def test(self): with self.test_session(): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1]) self.assertAllEqual(result.eval(), [5, 0, 0, 0, 0])
def testLarge(self): with self.cached_session(): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=17) with self.assertRaisesOpError("preserve_index out of range"): self.evaluate(result)
def testNegative(self): with self.cached_session(): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=-1) with self.assertRaisesOpError("Need preserve_index >= 0, got -1"): self.evaluate(result)
def testAttr(self): with self.cached_session(): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1], preserve_index=3) self.assertAllEqual(result.eval(), [0, 0, 0, 2, 0])
def test(self): with self.cached_session(): result = zero_out_op_3.zero_out([5, 4, 3, 2, 1]) self.assertAllEqual(result.eval(), [5, 0, 0, 0, 0])