Switch pytest import style.