thePR0M3TH3AN/seedPass
1
0
Fork 0
mirror of https://github.com/PR0M3TH3AN/SeedPass.git synced 2025-09-09 07:48:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #583 from PR0M3TH3AN/codex/update-test-cases-for-seed_prompt

Browse Source 
Update seed prompt tests
This commit is contained in:
thePR0M3TH3AN
2025-07-16 12:50:09 -04:00
committed by GitHub
parent 07f1843739 6b7815f28e
commit c6398b3c99
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/tests/test_seed_prompt.py
View File

@@ -37,9 +37,9 @@ def test_prompt_seed_words_valid(monkeypatch):
phrase = m.generate(strength=128) phrase = m.generate(strength=128)
words = phrase.split() words = phrase.split()
inputs = iter(words + ["y"] * len(words)) word_iter = iter(words)
monkeypatch.setattr(seed_prompt, "masked_input", lambda *_: next(inputs)) monkeypatch.setattr(seed_prompt, "masked_input", lambda *_: next(word_iter))
monkeypatch.setattr("builtins.input", lambda *_: next(inputs)) monkeypatch.setattr("builtins.input", lambda *_: "y")
result = seed_prompt.prompt_seed_words(len(words)) result = seed_prompt.prompt_seed_words(len(words))
assert result == phrase assert result == phrase
@@ -52,9 +52,9 @@ def test_prompt_seed_words_invalid_word(monkeypatch):
phrase = m.generate(strength=128) phrase = m.generate(strength=128)
words = phrase.split() words = phrase.split()
# Insert an invalid word for the first entry then the correct one # Insert an invalid word for the first entry then the correct one
inputs = iter(["invalid"] + [words[0]] + words[1:] + ["y"] * len(words)) inputs = iter(["invalid"] + words)
monkeypatch.setattr(seed_prompt, "masked_input", lambda *_: next(inputs)) monkeypatch.setattr(seed_prompt, "masked_input", lambda *_: next(inputs))
monkeypatch.setattr("builtins.input", lambda *_: next(inputs)) monkeypatch.setattr("builtins.input", lambda *_: "y")
result = seed_prompt.prompt_seed_words(len(words)) result = seed_prompt.prompt_seed_words(len(words))
assert result == phrase assert result == phrase