code: def test_serialization(self):
      data = torch.zeros((2, 2), dtype=torch.int32)
      out = save({"test": data})
  
      self.assertEqual(
          out,
          b'@\x00\x00\x00\x00\x00\x00\x00{"test":{"dtype":"I32","shape":[2,2],"data_offsets":[0,16]}}   '
          b" \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00",
      )
  
      save_file({"test": data}, "serialization.safetensors")
      out = open("serialization.safetensors", "rb").read()
      self.assertEqual(
          out,
          b'@\x00\x00\x00\x00\x00\x00\x00{"test":{"dtype":"I32","shape":[2,2],"data_offsets":[0,16]}}   '
          b" \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00",
      )
  
      data[1, 1] = 1
      out = save({"test": data})
  
      self.assertEqual(
          out,
          b'@\x00\x00\x00\x00\x00\x00\x00{"test":{"dtype":"I32","shape":[2,2],"data_offsets":[0,16]}}   '
          b" \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00",
      )
      save_file({"test": data}, "serialization.safetensors")
      out = open("serialization.safetensors", "rb").read()
      self.assertEqual(
          out,
          b'@\x00\x00\x00\x00\x00\x00\x00{"test":{"dtype":"I32","shape":[2,2],"data_offsets":[0,16]}}   '
          b" \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00",
      )
  
imports: import sys
  import unittest
  import torch
  from safetensors import safe_open
  from safetensors.torch import load, load_file, save, save_file
label_desc: Use image node_id 1 for calling the TorchTestCase obj's underlying member method code with example usage: obj.test_serialization() without return types