Update test.py

Make sure the tests assert something so that you can run nosetests on this (https://nose.readthedocs.org/en/latest/).