implement unit tests for simple heuristic