test: python infra_tests.py train: python infra_tests.py --train