code: def test_construction(self):
      sequential = nn.Sequential([nn.Dense(4), nn.Dense(2)])
      key1, key2 = random.split(random.key(0), 2)
      x = random.uniform(key1, (3, 1, 5))
      params = sequential.init(key2, x)
      output = sequential.apply(params, x)
      self.assertEqual(output.shape, (3, 1, 2))
  
imports: from typing import Any, Optional, Sequence
  import jax
  import numpy
  from absl.testing import absltest
  from jax import numpy
  from jax import random
  from flax import linen
label_desc: Use image node_id 1 for calling the SequentialTest obj's underlying member method code with example usage: obj.test_construction() without return types