def test_select_branch_room(self):
builder = LevelBuilder()
r1, r2, r3 = object(), object(), object()
builder.rooms = {1: r1, 11: r2, 111: r3}
selected = set()
for _ in xrange(32):
selected.add(builder.select_branch_room())
self.assertEquals(selected, set([r1, r2, r3]))
