spb/uqo-chat

1from app.sandbox.client import prefilter234def test_prefilter_blocks_hostile_code() -> None:5    for code in ["import subprocess", "import socket", "os.system('ls')", "__import__('os')",6                 "import requests", "eval('1')", "open('/etc/passwd')"]:7        assert prefilter(code) is not None, code8910def test_prefilter_allows_normal_code() -> None:11    assert prefilter("import pandas as pd\nprint(pd.DataFrame({'a':[1]}).sum())") is None12    assert prefilter("import numpy_financial as npf\nprint(npf.pmt(0.05/12, 300, -350000))") is None13