Add test cases for branches