put pytest import inside main() function