code: def test_max_vit(self):
      data = torch.rand(1, 3, 224, 224)
      if torch.cuda.is_available():
          data = data.cuda()
  
      model = create_model(model_name="max_vit_tiny")
      output = model(data)
      self.assertEqual(output.shape, (1, 1000))
  
imports: import unittest
  import torch
  from towhee.models.max_vit.max_vit import create_model
label_desc: Use image node_id 1 for calling the MaxVitUtilsTest obj's underlying member method code with example usage: obj.test_max_vit() without return types