add unit testing for brute_force