Write a function to find all words starting with 'a' or 'e' in a given string.
import re
def words_ae(text):
 list = re.findall("[ae]\w+", text)
 return list
<s>[INST]<<SYS>>Given a task and its code, generate test cases for the code.<</SYS>>Write a function to find all words starting with 'a' or 'e' in a given string.
import re
def words_ae(text):
 list = re.findall("[ae]\w+", text)
 return list[/INST]assert words_ae("python programe")==['ame']
assert words_ae("python programe language")==['ame','anguage']
assert words_ae("assert statement")==['assert', 'atement']</s>