Added my own expect framework to replace assert