Added a test setup helper function