write test for get_pk_param function

[Yousuf28]

#2 write a test for function to make sure it calculating mean correctly