Spaces:
Runtime error
Runtime error
File size: 1,880 Bytes
aade3e0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | from pathlib import Path
import unittest
ROOT = Path(__file__).resolve().parents[1]
def read_text(relative_path: str) -> str:
return (ROOT / relative_path).read_text(encoding="utf-8")
class ExternalSourcePolicyTest(unittest.TestCase):
def test_app_uses_phr00t_v23_single_file_source(self):
app = read_text("app.py")
self.assertNotIn("Sneak-Moose", app)
self.assertIn("Phr00t/Qwen-Image-Edit-Rapid-AIO", app)
self.assertIn("v23/Qwen-Rapid-AIO-NSFW-v23.safetensors", app)
self.assertIn("load_phr00t_rapid_transformer", app)
self.assertIn('PHR00T_TRANSFORMER_PREFIX = "model.diffusion_model."', app)
self.assertIn("safe_open", app)
self.assertNotIn("from_single_file", app)
def test_private_diagnostics_callback_is_removed(self):
app = read_text("app.py")
self.assertNotIn("QUALITY_ENHANCEMENT_URL", app)
self.assertNotIn("QUALITY_ENHANCEMENT_TOKEN", app)
self.assertNotIn("requests.post", app)
self.assertNotIn("_emit_diagnostics", app)
def test_third_party_video_space_is_not_hardcoded(self):
app = read_text("app.py")
self.assertNotIn("multimodalart/wan-2-2-first-last-frame", app)
self.assertIn('VIDEO_SPACE_ID = os.environ.get("VIDEO_SPACE_ID", "").strip()', app)
self.assertIn("Client(VIDEO_SPACE_ID)", app)
def test_no_external_logo_image_or_sneak_moose_docs(self):
relevant_text = "\n".join(
read_text(path)
for path in ("app.py", "README.md", "AGENT.md", "CLAUDE.md")
)
self.assertNotIn("Sneak-Moose", relevant_text)
self.assertNotIn("qianwen-res.oss-cn-beijing.aliyuncs.com", relevant_text)
self.assertNotIn("Rapid-AIO v18", relevant_text)
if __name__ == "__main__":
unittest.main()
|