code: def normal_test(func, shape=(20, 23), alpha=0.05):
      return equal_distribution(
          func,
          numpy_func=lambda x: np.random.randn(*x),
          shape=shape,
          alpha=alpha,
      )
  
imports: import math
  import unittest
  import numpy
  import torch
  from tinygrad import nn, dtypes, Tensor
  from functools import partial
label_desc: Use image node_id 4 for calling a global function with example usage: normal_test(func, shape, alpha) and returns: equal_distribution