def test_can_use_intermediate_rules():
rule_set = RuleSet()
smell = NonTerminal("SMELL")
rule_set.add(sentence, "You smell of ", smell)
rule_set.add(smell, "elderberries")
rule_set.add(smell, "dogfood")
assert_equals("You smell of elderberries", generate(rule_set, StaticSelector(0, 0)))
assert_equals("You smell of dogfood", generate(rule_set, StaticSelector(0, 1)))
def test_can_generate_all_sentences_up_to_maximum_depth():
rule_set = RuleSet()
rule_set.add(sentence, "B")
rule_set.add(sentence, "A ", sentence)
assert_equals(["B", "A B", "A A B"], generate_all(rule_set, 3))
def test_can_generate_all_sentences():
rule_set = RuleSet()
rule_set.add(sentence, "I hate you!")
smell = NonTerminal("SMELL")
rude_adj = NonTerminal("RUDE_ADJ")
pet = NonTerminal("PET")
rule_set.add(sentence, "You smell of ", smell)
rule_set.add(smell, "elderberries")
rule_set.add(smell, "dogfood")
rule_set.add(sentence, "You're as ", rude_adj, " as my ", pet)
rule_set.add(rude_adj, "ugly")
rule_set.add(rude_adj, "stupid")
rule_set.add(pet, "rat")
rule_set.add(pet, "dog")
assert_equals(["I hate you!", "You smell of elderberries", "You smell of dogfood",
"You're as ugly as my rat", "You're as ugly as my dog",
"You're as stupid as my rat", "You're as stupid as my dog"], generate_all(rule_set))
def test_can_generate_empty_sentence():
rule_set = RuleSet()
rule_set.add(sentence)
assert_equals("", generate(rule_set, StaticSelector(0)))
def test_resorts_to_picking_sentence_randomly_from_all_sentences_if_non_terminal_cannot_be_expanded():
rule_set = RuleSet()
rule_set.add(sentence, NonTerminal("RUDE_WORD"))
rule_set.add(sentence, "Go away!")
assert_equals("Go away!", generate(rule_set, StaticSelector(0, 0)));
def test_resorts_to_picking_sentence_randomly_from_all_sentences_if_maximum_depth_is_reached():
rule_set = RuleSet()
rule_set.add(sentence, "B")
rule_set.add(sentence, "A ", sentence)
assert_equals("A A B", generate(rule_set, StaticSelector(1, 1, 1, 2), 3));
def test_selector_is_used_to_pick_rule_when_there_are_many_possible_rules():
rule_set = RuleSet()
rule_set.add(sentence, "I hate you!")
rule_set.add(sentence, "Go to hell!")
assert_equals("I hate you!", generate(rule_set, StaticSelector(0)))
assert_equals("Go to hell!", generate(rule_set, StaticSelector(1)))
def test_adding_rule_to_convert_sentence_to_terminal():
rule_set = RuleSet()
rule_set.add(sentence, "I hate you!")
assert_equals("I hate you!", generate(rule_set, StaticSelector(0)))
