Best Python code snippet using avocado_python
tridiagonal_solve_op_test.py
Source:tridiagonal_solve_op_test.py
...401 })402 # Invalid input shapes403 @flags(FLAG_NO_PARAMETERIZATION)404 def testInvalidShapesCompactFormat(self):405 def test_raises(diags_shape, rhs_shape):406 self._assertRaises(_tf_ones(diags_shape), _tf_ones(rhs_shape), "compact")407 test_raises((5, 4, 4), (5, 4))408 test_raises((5, 3, 4), (4, 5))409 test_raises((5, 3, 4), (5))410 test_raises((5), (5, 4))411 @flags(FLAG_NO_PARAMETERIZATION)412 def testInvalidShapesSequenceFormat(self):413 def test_raises(diags_tuple_shapes, rhs_shape):414 diagonals = tuple(_tf_ones(shape) for shape in diags_tuple_shapes)415 self._assertRaises(diagonals, _tf_ones(rhs_shape), "sequence")416 test_raises(((5, 4), (5, 4)), (5, 4))417 test_raises(((5, 4), (5, 4), (5, 6)), (5, 4))418 test_raises(((5, 3), (5, 4), (5, 6)), (5, 4))419 test_raises(((5, 6), (5, 4), (5, 3)), (5, 4))420 test_raises(((5, 4), (7, 4), (5, 4)), (5, 4))421 test_raises(((5, 4), (7, 4), (5, 4)), (3, 4))422 @flags(FLAG_NO_PARAMETERIZATION)423 def testInvalidShapesMatrixFormat(self):424 def test_raises(diags_shape, rhs_shape):425 self._assertRaises(_tf_ones(diags_shape), _tf_ones(rhs_shape), "matrix")426 test_raises((5, 4, 7), (5, 4))427 test_raises((5, 4, 4), (3, 4))428 test_raises((5, 4, 4), (5, 3))429 # Tests with placeholders430 def _testWithPlaceholders(self,431 diags_shape,432 rhs_shape,433 diags_feed,434 rhs_feed,435 expected,436 diags_format="compact"):437 if context.executing_eagerly():438 return439 diags = array_ops.placeholder(dtypes.float64, shape=diags_shape)440 rhs = array_ops.placeholder(dtypes.float64, shape=rhs_shape)441 x = linalg_impl.tridiagonal_solve(442 diags, rhs, diags_format, partial_pivoting=self.pivoting)...
tridiagonal_solve_ops_test.py
Source:tridiagonal_solve_ops_test.py
...394 with self.assertRaises(ValueError):395 linalg_impl.tridiagonal_solve(diags, rhs, diags_format)396 # Invalid input shapes397 def testInvalidShapesCompactFormat(self):398 def test_raises(diags_shape, rhs_shape):399 self._assertRaises(_tf_ones(diags_shape), _tf_ones(rhs_shape), "compact")400 test_raises((5, 4, 4), (5, 4))401 test_raises((5, 3, 4), (4, 5))402 test_raises((5, 3, 4), (5))403 test_raises((5), (5, 4))404 def testInvalidShapesSequenceFormat(self):405 def test_raises(diags_tuple_shapes, rhs_shape):406 diagonals = tuple(_tf_ones(shape) for shape in diags_tuple_shapes)407 self._assertRaises(diagonals, _tf_ones(rhs_shape), "sequence")408 test_raises(((5, 4), (5, 4)), (5, 4))409 test_raises(((5, 4), (5, 4), (5, 6)), (5, 4))410 test_raises(((5, 3), (5, 4), (5, 6)), (5, 4))411 test_raises(((5, 6), (5, 4), (5, 3)), (5, 4))412 test_raises(((5, 4), (7, 4), (5, 4)), (5, 4))413 test_raises(((5, 4), (7, 4), (5, 4)), (3, 4))414 def testInvalidShapesMatrixFormat(self):415 def test_raises(diags_shape, rhs_shape):416 self._assertRaises(_tf_ones(diags_shape), _tf_ones(rhs_shape), "matrix")417 test_raises((5, 4, 7), (5, 4))418 test_raises((5, 4, 4), (3, 4))419 test_raises((5, 4, 4), (5, 3))420 # Tests involving placeholder with an unknown dimension are all skipped.421 # Dimensions have to be all known statically.422if __name__ == "__main__":...
Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Get 100 minutes of automation test minutes FREE!!