Add audio separation web app and verified recovery sources

test_separator_sources.py

@@ -0,0 +1,19 @@
+import hashlib,json,tempfile,unittest
+from pathlib import Path
+from unittest.mock import patch
+from separator_runtime import SOURCES,MODELS,verify_package,package_files
+class Tests(unittest.TestCase):
+ def test_demucs_ensemble_complete_and_sources_https(self):
+  self.assertEqual(set(SOURCES),set(MODELS))
+  self.assertEqual(len(package_files('htdemucs_ft.yaml')),5)
+  self.assertEqual(len(package_files('htdemucs_6s.yaml')),2)
+  for recipe in SOURCES.values():
+   for f in recipe['files']:
+    self.assertEqual(len(f['sha256']),64)
+    self.assertTrue(all(u.startswith('https://') for u in f['urls']))
+ def test_wrong_or_missing_weight_rejected(self):
+  with tempfile.TemporaryDirectory() as d:
+   recipe={'x':{'files':[{'file':'weight','sha256':hashlib.sha256(b'valid').hexdigest()}]}}
+   with patch.dict(SOURCES,recipe):
+    self.assertFalse(verify_package(d,'x'));Path(d,'weight').write_bytes(b'wrong');self.assertFalse(verify_package(d,'x'))
+    Path(d,'weight').write_bytes(b'valid');self.assertTrue(verify_package(d,'x'))