TST: Add tests for branching_factor