docstr: Test assigning invalid shape and type
docstr_tok: ["Test","assigning","invalid","shape","and","type"]
code: def test_assign_invalid_shape_and_type(self):
      """Test assigning invalid shape and type"""
      t = objax.TrainVar(jn.zeros(2))
      t.assign(jn.ones(2))
      with self.assertRaises(AssertionError):
          t.assign(jn.zeros(3))
      with self.assertRaises(AssertionError):
          t.assign([0, 1])
      t.assign(jn.ones(3), check=False)
  
imports: import unittest
  import numpy
  import jax.numpy
  import objax
label_desc: Use image node_id 7 for calling the TestVariable obj's underlying member method code with example usage: obj.test_assign_invalid_shape_and_type() without return types