Add helper functions in test_test.go