def test_get_tensor_attribute_tensor(module: DynamicBufferModule) -> None:
        """Test the get_tensor_attribute method with a tensor field."""
        tensor_attribute = module.get_tensor_attribute("tensor_attribute")
        assert isinstance(tensor_attribute, torch.Tensor)
        assert torch.equal(tensor_attribute, module.tensor_attribute)
Test the get_tensor_attribute method with a tensor field.
tests/unit/models/components/base/test_dynamic_buffer_mixin.py