Make tassel test script more generic