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]))