Add pie tests