def test_broken_linked_options(self):
choice_1 = facts.Choice(uid="choice_1")
choice_2 = facts.Choice(uid="choice_2")
o_1_2 = facts.Option(state_from=choice_1.uid, state_to=choice_2.uid, type="o", markers=())
o_1_f1 = facts.Option(state_from=choice_1.uid, state_to="st_finish_1", type="o", markers=())
o_2_f1 = facts.Option(state_from=choice_2.uid, state_to="st_finish_1", type="o", markers=())
o_2_f2 = facts.Option(state_from=choice_2.uid, state_to="st_finish_2", type="o", markers=())
facts_list = [
facts.Start(uid="start", type="test", nesting=0),
facts.Jump(state_from="start", state_to=choice_1.uid),
choice_1,
o_1_f1,
facts.Finish(uid="st_finish_1", results={}, nesting=0, start="start"),
]
self.kb += facts_list
broken = [
choice_2,
o_1_2,
o_2_f1,
o_2_f2,
facts.Finish(uid="st_finish_2", results={}, nesting=1, start="start"),
]
self.kb += broken
self.kb += [facts.OptionsLink(options=[o_1_2.uid, o_2_f2.uid])]
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
self.check_not_in_knowledge_base(self.kb, broken)
def test_wrong_answers_number__more_then_2_answers(self):
self.kb += facts.Answer(state_from='state_1',
state_to='finish_1',
condition=True)
transformators.remove_broken_states(self.kb)
self.check_not_in_knowledge_base(
self.kb, [self.question, self.answer_1, self.answer_2])
def _create_random_quest_for_hero(hero_info, start_quests, without_restrictions=False):
knowledge_base = get_knowledge_base(hero_info, without_restrictions=without_restrictions)
selector = Selector(knowledge_base, QUESTS_BASE, social_connection_probability=c.QUESTS_SOCIAL_CONNECTIONS_FRACTION)
hero_uid = uids.hero(hero_info.id)
quests_facts = selector.create_quest_from_place(nesting=0,
initiator_position=selector.place_for(objects=(hero_uid,)),
allowed=start_quests,
excluded=[],
tags=('can_start', ))
knowledge_base += quests_facts
transformators.activate_events(knowledge_base) # TODO: after remove restricted states
transformators.remove_restricted_states(knowledge_base)
transformators.remove_broken_states(knowledge_base) # MUST be called after all graph changes
transformators.determine_default_choices(knowledge_base, preferred_markers=hero_info.prefered_quest_markers) # MUST be called after all graph changes and on valid graph
transformators.remove_unused_actors(knowledge_base)
knowledge_base.validate_consistency(WORLD_RESTRICTIONS)
knowledge_base.validate_consistency(QUEST_RESTRICTIONS)
return knowledge_base
def _create_random_quest_for_hero(hero_info,
start_quests,
without_restrictions=False):
knowledge_base = get_knowledge_base(
hero_info, without_restrictions=without_restrictions)
selector = Selector(knowledge_base,
QUESTS_BASE,
social_connection_probability=0)
hero_uid = uids.hero(hero_info.id)
quests_facts = selector.create_quest_from_place(
nesting=0,
initiator_position=selector.place_for(objects=(hero_uid, )),
allowed=start_quests,
excluded=[],
tags=('can_start', ))
knowledge_base += quests_facts
transformators.activate_events(
knowledge_base) # TODO: after remove restricted states
transformators.remove_restricted_states(knowledge_base)
transformators.remove_broken_states(
knowledge_base) # MUST be called after all graph changes
transformators.determine_default_choices(
knowledge_base, preferred_markers=hero_info.prefered_quest_markers
) # MUST be called after all graph changes and on valid graph
transformators.remove_unused_actors(knowledge_base)
knowledge_base.validate_consistency(WORLD_RESTRICTIONS)
knowledge_base.validate_consistency(QUEST_RESTRICTIONS)
return knowledge_base
def test_no_broken_states(self):
facts_list = [ facts.Start(uid='start', type='test', nesting=0),
facts.Finish(uid='st_finish', results={}, nesting=0, start='start'),
facts.Jump(state_from='start', state_to='st_finish') ]
self.kb += facts_list
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
def test_wrong_answers_structure__false(self):
self.kb -= self.answer_1
self.kb += facts.Answer(state_from='state_1',
state_to='state_2',
condition=False)
transformators.remove_broken_states(self.kb)
self.check_not_in_knowledge_base(
self.kb, [self.question, self.answer_1, self.answer_2])
def test_no_broken_states(self):
facts_list = [
facts.Start(uid="start", type="test", nesting=0),
facts.Finish(uid="st_finish", results={}, nesting=0, start="start"),
facts.Jump(state_from="start", state_to="st_finish"),
]
self.kb += facts_list
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
def test_no_broken_states(self):
facts_list = [
facts.Start(uid='start', type='test', nesting=0),
facts.Finish(uid='st_finish', results={}, nesting=0,
start='start'),
facts.Jump(state_from='start', state_to='st_finish')
]
self.kb += facts_list
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
def test_broken_linked_options(self):
choice_1 = facts.Choice(uid='choice_1')
choice_2 = facts.Choice(uid='choice_2')
o_1_2 = facts.Option(state_from=choice_1.uid,
state_to=choice_2.uid,
type='o',
markers=())
o_1_f1 = facts.Option(state_from=choice_1.uid,
state_to='st_finish_1',
type='o',
markers=())
o_2_f1 = facts.Option(state_from=choice_2.uid,
state_to='st_finish_1',
type='o',
markers=())
o_2_f2 = facts.Option(state_from=choice_2.uid,
state_to='st_finish_2',
type='o',
markers=())
facts_list = [
facts.Start(uid='start', type='test', nesting=0),
facts.Jump(state_from='start', state_to=choice_1.uid), choice_1,
o_1_f1,
facts.Finish(uid='st_finish_1',
results={},
nesting=0,
start='start')
]
self.kb += facts_list
broken = [
choice_2, o_1_2, o_2_f1, o_2_f2,
facts.Finish(uid='st_finish_2',
results={},
nesting=1,
start='start')
]
self.kb += broken
self.kb += [facts.OptionsLink(options=[o_1_2.uid, o_2_f2.uid])]
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
self.check_not_in_knowledge_base(self.kb, broken)
def test_broken_jumps(self):
facts_list = [ facts.Start(uid='start', type='test', nesting=0),
facts.Finish(uid='st_finish', results={}, nesting=0, start='start'),
facts.Jump(state_from='start', state_to='st_finish') ]
self.kb += facts_list
broken_jump_from = facts.Jump(state_from='st_broken_state', state_to='st_finish')
broken_jump_to = facts.Jump(state_from='start', state_to='st_broken_state')
self.kb += (broken_jump_from, broken_jump_to)
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
self.assertFalse(broken_jump_from.uid in self.kb)
self.assertFalse(broken_jump_to.uid in self.kb)
def test_broken_jumps(self):
facts_list = [
facts.Start(uid="start", type="test", nesting=0),
facts.Finish(uid="st_finish", results={}, nesting=0, start="start"),
facts.Jump(state_from="start", state_to="st_finish"),
]
self.kb += facts_list
broken_jump_from = facts.Jump(state_from="st_broken_state", state_to="st_finish")
broken_jump_to = facts.Jump(state_from="start", state_to="st_broken_state")
self.kb += (broken_jump_from, broken_jump_to)
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
self.assertFalse(broken_jump_from.uid in self.kb)
self.assertFalse(broken_jump_to.uid in self.kb)
def test_broken_jumps(self):
facts_list = [
facts.Start(uid='start', type='test', nesting=0),
facts.Finish(uid='st_finish', results={}, nesting=0,
start='start'),
facts.Jump(state_from='start', state_to='st_finish')
]
self.kb += facts_list
broken_jump_from = facts.Jump(state_from='st_broken_state',
state_to='st_finish')
broken_jump_to = facts.Jump(state_from='start',
state_to='st_broken_state')
self.kb += (broken_jump_from, broken_jump_to)
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
self.assertFalse(broken_jump_from.uid in self.kb)
self.assertFalse(broken_jump_to.uid in self.kb)
def test_success(self):
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(
self.kb, [self.question, self.answer_1, self.answer_2])
def create_quest():
# формируем список заданий для генерации
qb = QuestsBase()
qb += [
Spying, Hunt, Hometown, SearchSmith, Delivery, Caravan, CollectDebt,
HelpFriend, InterfereEnemy, Help
]
kb = KnowledgeBase()
# описываем мир
kb += [
facts.Hero(uid='hero'), # наш герой
facts.Place(uid='place_1', terrains=(
1,
)), # есть место с идентификатором place_1 и типами ландшафта 1,
facts.Place(uid='place_2', terrains=(0, )),
facts.Place(uid='place_3', terrains=(0, )),
facts.Place(uid='place_4', terrains=(1, )),
facts.Place(uid='place_5', terrains=(2, )),
facts.Place(uid='place_6', terrains=(1, )),
facts.Place(uid='place_7', terrains=(2, )),
facts.Place(uid='place_8', terrains=(2, )),
facts.Place(uid='place_9', terrains=(1, )),
facts.Place(uid='place_10', terrains=(2, )),
facts.Person(
uid='person_1', profession=PROFESSION.NONE
), # есть персонаж с идентификатором perons_1 и без профессии
facts.Person(uid='person_2', profession=PROFESSION.BLACKSMITH),
facts.Person(uid='person_3', profession=PROFESSION.ROGUE),
facts.Person(uid='person_4', profession=PROFESSION.NONE),
facts.Person(uid='person_5', profession=PROFESSION.NONE),
facts.Person(uid='person_6', profession=PROFESSION.NONE),
facts.Person(uid='person_7', profession=PROFESSION.NONE),
facts.Person(uid='person_8', profession=PROFESSION.NONE),
facts.Person(uid='person_9', profession=PROFESSION.NONE),
facts.Person(uid='person_10', profession=PROFESSION.NONE),
facts.LocatedIn(
object='person_1',
place='place_1'), # персонаж person_1 находится в place_1
facts.LocatedIn(object='person_2', place='place_2'),
facts.LocatedIn(object='person_3', place='place_3'),
facts.LocatedIn(object='person_4', place='place_4'),
facts.LocatedIn(object='person_5', place='place_5'),
facts.LocatedIn(object='person_6', place='place_6'),
facts.LocatedIn(object='person_7', place='place_7'),
facts.LocatedIn(object='person_8', place='place_8'),
facts.LocatedIn(object='person_9', place='place_9'),
facts.LocatedIn(object='person_10', place='place_10'),
facts.LocatedIn(object='hero',
place='place_1'), # герой находится в place_1
facts.Mob(
uid='mob_1', terrains=(0, )
), # есть монстр, обитающий на территориях с идентификатором 0 (для задания на охоту)
facts.PreferenceMob(
object='hero',
mob='mob_1'), # герой любит охотиться на монстра mob_1
facts.PreferenceHometown(
object='hero',
place='place_2'), # герой считате родным место place_2
facts.PreferenceFriend(object='hero',
person='person_4'), # герой дружит с person_4
facts.PreferenceEnemy(object='hero',
person='person_5'), # герой враждует с person_5
# указываем, что обновление экипировки стоит 777 монет (для задания SearchSmith)
# facts.HasMoney(object='hero', money=888), # если этот факт раскоментировать то в этом задании герой купит экипировку, а не пойдёт делать задание кузнеца
facts.UpgradeEquipmentCost(money=777),
facts.OnlyGoodBranches(object='place_2'), # не вредить месту place_2
facts.OnlyGoodBranches(
object='person_4'), # не вредить персонажу person_4
facts.OnlyBadBranches(object='person_5')
] # не помогать персонажу person_5
kb.validate_consistency(
WORLD_RESTRICTIONS) # проверяем ограничения на мир,
selector = Selector(kb, qb)
# создаём квест (получаем список фактов)
quests_facts = selector.create_quest_from_place(
nesting=0, initiator_position=kb['place_1'], tags=('can_start', ))
kb += quests_facts
transformators.activate_events(
kb
) # активируем события (из нескольких вершин графа оставляем одну, остальные удаляем)
transformators.remove_restricted_states(
kb
) # удаляем состояния, в которые нельзя переходить (например, которые вредят тому, кому вредить нельщя)
transformators.remove_broken_states(
kb
) # чистим граф задания от разрушений, вызванных предыдущими действиями
transformators.determine_default_choices(
kb) # определяем выборы по умолчанию на развилках
kb.validate_consistency(WORLD_RESTRICTIONS) # ещё раз проверяем мир
kb.validate_consistency(
QUEST_RESTRICTIONS
) # проверяем граф задания (вдруг полностью разрушен)
return kb
def test_wrong_answers_number__less_then_2_answers(self):
self.kb -= self.answer_1
transformators.remove_broken_states(self.kb)
self.check_not_in_knowledge_base(self.kb, [self.question, self.answer_1, self.answer_2])
def test_success(self):
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, [self.question, self.answer_1, self.answer_2])
def test_wrong_answers_structure__true(self):
self.kb -= self.answer_2
self.kb += facts.Answer(state_from='state_1', state_to='finish_1', condition=True)
transformators.remove_broken_states(self.kb)
self.check_not_in_knowledge_base(self.kb, [self.question, self.answer_1, self.answer_2])
def test_wrong_answers_number__more_then_2_answers(self):
self.kb += facts.Answer(state_from="state_1", state_to="finish_1", condition=True)
transformators.remove_broken_states(self.kb)
self.check_not_in_knowledge_base(self.kb, [self.question, self.answer_1, self.answer_2])
def test_wrong_answers_number__less_then_2_answers(self):
self.kb -= self.answer_1
transformators.remove_broken_states(self.kb)
self.check_not_in_knowledge_base(
self.kb, [self.question, self.answer_1, self.answer_2])
def test_wrong_answers_structure__false(self):
self.kb -= self.answer_1
self.kb += facts.Answer(state_from="state_1", state_to="state_2", condition=False)
transformators.remove_broken_states(self.kb)
self.check_not_in_knowledge_base(self.kb, [self.question, self.answer_1, self.answer_2])
def create_quest():
# формируем список заданий для генерации
qb = QuestsBase()
qb += [Spying, Hunt, Hometown, SearchSmith, Delivery, Caravan, CollectDebt, HelpFriend, InterfereEnemy, Help]
kb = KnowledgeBase()
# описываем мир
kb += [ facts.Hero(uid='hero'), # наш герой
facts.Place(uid='place_1', terrains=(1,)), # есть место с идентификатором place_1 и типами ландшафта 1,
facts.Place(uid='place_2', terrains=(0,)),
facts.Place(uid='place_3', terrains=(0,)),
facts.Place(uid='place_4', terrains=(1,)),
facts.Place(uid='place_5', terrains=(2,)),
facts.Place(uid='place_6', terrains=(1,)),
facts.Place(uid='place_7', terrains=(2,)),
facts.Place(uid='place_8', terrains=(2,)),
facts.Place(uid='place_9', terrains=(1,)),
facts.Place(uid='place_10', terrains=(2,)),
facts.Person(uid='person_1', profession=PROFESSION.NONE), # есть персонаж с идентификатором perons_1 и без профессии
facts.Person(uid='person_2', profession=PROFESSION.BLACKSMITH),
facts.Person(uid='person_3', profession=PROFESSION.ROGUE),
facts.Person(uid='person_4', profession=PROFESSION.NONE),
facts.Person(uid='person_5', profession=PROFESSION.NONE),
facts.Person(uid='person_6', profession=PROFESSION.NONE),
facts.Person(uid='person_7', profession=PROFESSION.NONE),
facts.Person(uid='person_8', profession=PROFESSION.NONE),
facts.Person(uid='person_9', profession=PROFESSION.NONE),
facts.Person(uid='person_10', profession=PROFESSION.NONE),
facts.LocatedIn(object='person_1', place='place_1'), # персонаж person_1 находится в place_1
facts.LocatedIn(object='person_2', place='place_2'),
facts.LocatedIn(object='person_3', place='place_3'),
facts.LocatedIn(object='person_4', place='place_4'),
facts.LocatedIn(object='person_5', place='place_5'),
facts.LocatedIn(object='person_6', place='place_6'),
facts.LocatedIn(object='person_7', place='place_7'),
facts.LocatedIn(object='person_8', place='place_8'),
facts.LocatedIn(object='person_9', place='place_9'),
facts.LocatedIn(object='person_10', place='place_10'),
facts.LocatedIn(object='hero', place='place_1'), # герой находится в place_1
facts.Mob(uid='mob_1', terrains=(0,)), # есть монстр, обитающий на территориях с идентификатором 0 (для задания на охоту)
facts.PreferenceMob(object='hero', mob='mob_1'), # герой любит охотиться на монстра mob_1
facts.PreferenceHometown(object='hero', place='place_2'), # герой считате родным место place_2
facts.PreferenceFriend(object='hero', person='person_4'), # герой дружит с person_4
facts.PreferenceEnemy(object='hero', person='person_5'), # герой враждует с person_5
# указываем, что обновление экипировки стоит 777 монет (для задания SearchSmith)
# facts.HasMoney(object='hero', money=888), # если этот факт раскоментировать то в этом задании герой купит экипировку, а не пойдёт делать задание кузнеца
facts.UpgradeEquipmentCost(money=777),
facts.OnlyGoodBranches(object='place_2'), # не вредить месту place_2
facts.OnlyGoodBranches(object='person_4'), # не вредить персонажу person_4
facts.OnlyBadBranches(object='person_5') ] # не помогать персонажу person_5
kb.validate_consistency(WORLD_RESTRICTIONS) # проверяем ограничения на мир,
selector = Selector(kb, qb)
# создаём квест (получаем список фактов)
quests_facts = selector.create_quest_from_place(nesting=0,
initiator_position=kb['place_1'],
tags=('can_start', ))
kb += quests_facts
transformators.activate_events(kb) # активируем события (из нескольких вершин графа оставляем одну, остальные удаляем)
transformators.remove_restricted_states(kb) # удаляем состояния, в которые нельзя переходить (например, которые вредят тому, кому вредить нельщя)
transformators.remove_broken_states(kb) # чистим граф задания от разрушений, вызванных предыдущими действиями
transformators.determine_default_choices(kb) # определяем выборы по умолчанию на развилках
kb.validate_consistency(WORLD_RESTRICTIONS) # ещё раз проверяем мир
kb.validate_consistency(QUEST_RESTRICTIONS) # проверяем граф задания (вдруг полностью разрушен)
return kb
