Add tests for chi_square_p_value function