add some simple tests
make sure it works with python2