A simple python testing wrapper