code: def test_module_random_walk_pe(idtype):
      transform = dgl.RandomWalkPE(2, "rwpe")
      g = dgl.graph(
          ([0, 1, 1], [1, 1, 0]), idtype=idtype, device=F.ctx()
      )
      new_g = transform(g)
      tgt = F.copy_to(F.tensor([[0.0, 0.5], [0.5, 0.75]]), g.device)
      assert F.allclose(new_g.ndata["rwpe"], tgt)
  
imports: import math
  import os
  import unittest
  import backend
  import dgl
  import dgl.function
  import dgl.partition
  import networkx
  import numpy
  import pytest
  from scipy import sparse
  from utils import parametrize_idtype
  from utils.graph_cases import get_cases
label_desc: Use image node_id 49 for calling a global function with example usage: test_module_random_walk_pe(idtype) without return types