Skip to content

Commit 193fcd0

Browse files
committed
fix: updated correct imports for monkeypatch
1 parent 756787f commit 193fcd0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_decompiler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,16 +4,16 @@
44

55
def test_decompile(monkeypatch):
66
disasm = "l1\nl2\nl3\nl4\n"
7-
monkeypatch.setattr("decompile_llm.decompiler.disassemble", lambda path: (disasm, None))
7+
monkeypatch.setattr("decompilellm.decompiler.disassemble", lambda path: (disasm, None))
88

99
calls = []
1010

1111
def fake_call_llm(api_key, model, system_message, prompt, provider, stream_output, output_file_handle=None, timeout=None, temperature=0.5, top_p=1.0, reasoning_effort=None):
1212
calls.append(prompt)
1313
return f"code{len(calls)}", None
1414

15-
monkeypatch.setattr("decompile_llm.decompiler.call_llm", fake_call_llm)
16-
monkeypatch.setattr("decompile_llm.decompiler.get_token_count", lambda text, model, provider: 1)
15+
monkeypatch.setattr("decompilellm.decompiler.call_llm", fake_call_llm)
16+
monkeypatch.setattr("decompilellm.decompiler.get_token_count", lambda text, model, provider: 1)
1717

1818
args = argparse.Namespace(
1919
pyc_file="dummy.pyc",

0 commit comments

Comments
 (0)