code: def test_odd_dtype(self):
      data = {
          "test": torch.zeros((2, 2), dtype=torch.bfloat16),
          "test2": torch.zeros((2, 2), dtype=torch.float16),
          "test3": torch.zeros((2, 2), dtype=torch.bool),
      }
      local = "./tests/data/out_safe_pt_mmap_small.safetensors"
  
      save_file(data, local)
      reloaded = load_file(local)
      self.assertTrue(torch.equal(data["test"], reloaded["test"]))
      self.assertTrue(torch.equal(data["test2"], reloaded["test2"]))
      self.assertTrue(torch.equal(data["test3"], reloaded["test3"]))
  
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 2 for calling the TorchTestCase obj's underlying member method code with example usage: obj.test_odd_dtype() without return types