Implement some primitive tests, fix division bug in get_infogain