Remove tests for removed code from learn.py