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!!
