First unit tests for budget