Update unit test for deep learner