have basic setup of testing with __init__.py file, need to separate into different files