Refactored tests to get rid of init_test.py. Resolves #49.