From 66c333c88e0c8a5e4844435c18effad9889bcaf5 Mon Sep 17 00:00:00 2001 From: Hykilpikonna Date: Sat, 23 Oct 2021 01:28:06 -0400 Subject: [PATCH] [+] A3 P2 Q3 1/2 --- assignments/a3/a3_part2.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/assignments/a3/a3_part2.py b/assignments/a3/a3_part2.py index 2aed7ba..e867390 100644 --- a/assignments/a3/a3_part2.py +++ b/assignments/a3/a3_part2.py @@ -64,6 +64,7 @@ def choose_from_keys(transitions: dict[str, list[str]]) -> str: Preconditions: - transitions != {} """ + return random.choice(list(transitions.keys())) def choose_from_follow_list(key: str, transitions: dict[str, list[str]]) -> str: @@ -77,6 +78,13 @@ def choose_from_follow_list(key: str, transitions: dict[str, list[str]]) -> str: - key in transitions - transitions[key] != [] """ + word = random.choice(transitions[key]) + transitions[key].remove(word) + + if len(transitions[key]) == 0: + del transitions[key] + + return word def generate_text_owc(count: int, transitions: dict[str, list[str]]) -> str: