using helper in test/main.py