After this PR is done I'm planning to move the bandits.py file to the tests subdir.