Add test for string generator