class_docstr: Test AssertFunction layer.
class_docstr_tok: ["Test","AssertFunction","layer","."]
code: def test_simple_fail(self):
      layer = tl.AssertFunction("abc->cba", tl.Dropout(rate=0.1))
      x = np.ones((2, 5, 20))
      with self.assertRaises(tl.LayerError):
          layer(x)
  
imports: from absl.testing import absltest
  import numpy
  import trax.layers
label_desc: Use image node_id 2 for calling the AssertFunctionTest obj's underlying member method code with example usage: obj.test_simple_fail() without return types