def test_buffer_list_device_placement(module: BufferListModule) -> None:
        """Test if the device of the buffer list is updated with the module.

        Note: This test will only run on a GPU-enabled device.
        """
        module.cuda()
        assert all(tensor.is_cuda for tensor in module.tensor_list)
        module.cpu()
        assert all(tensor.is_cpu for tensor in module.tensor_list)
Test if the device of the buffer list is updated with the module.

        Note: This test will only run on a GPU-enabled device.
        
tests/unit/models/components/base/test_buffer_list_mixin.py