switch out function tests for unittest tests